Skip to content

Commit 02cf084

Browse files
committed
GPU: Enable matLut with GPU Track Model in Track Refit
1 parent dfada64 commit 02cf084

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

GPU/GPUTracking/Refit/GPUTrackingRefit.cxx

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,7 @@ GPUd() void GPUTrackingRefit::initProp<GPUTPCGMPropagator>(GPUTPCGMPropagator& p
8282
prop.SetFitInProjections(mPparam->rec.fitInProjections != 0);
8383
prop.SetPropagateBzOnly(false);
8484
prop.SetPolynomialField(&mPparam->polynomialField);
85+
prop.SetMatLUT(mPmatLUT);
8586
}
8687

8788
template <>
@@ -415,6 +416,7 @@ void GPUTrackingRefit::SetPtrsFromGPUConstantMem(const GPUConstantMem* v, MEM_CO
415416
mPclusterNative = v->ioPtrs.clustersNative;
416417
mPtrackHits = v->ioPtrs.mergedTrackHits;
417418
mPfastTransform = v->calibObjects.fastTransform;
419+
mPmatLUT = v->calibObjects.matLUT;
418420
mPparam = p ? p : &v->param;
419421
}
420422

GPU/GPUTracking/Refit/GPUTrackingRefit.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ using TrackParCov = TrackParametrizationWithError<float>;
3131
namespace o2::base
3232
{
3333
class Propagator;
34+
class MatLayerCylSet;
3435
} // namespace o2::base
3536
namespace o2::tpc
3637
{
@@ -89,6 +90,7 @@ class GPUTrackingRefit
8990
private:
9091
const unsigned char* mPclusterState = nullptr; // Ptr to shared cluster state
9192
const o2::base::Propagator* mPpropagator = nullptr; // Ptr to propagator for TrackParCov track model
93+
const o2::base::MatLayerCylSet* mPmatLUT = nullptr; // Ptr to material LUT
9294
const o2::tpc::ClusterNativeAccess* mPclusterNative = nullptr; // Ptr to cluster native access structure
9395
const GPUTPCGMMergedTrackHit* mPtrackHits = nullptr; // Ptr to hits for GPUTPCGMMergedTrack tracks
9496
const unsigned int* mPtrackHitReferences = nullptr; // Ptr to hits for TrackTPC tracks

0 commit comments

Comments
 (0)