GPU: Add gpu-reconstruction-prepare device, forwarding the pointers t…

…o gpu-reconstruction
AliceO2Group · Sep 22, 2023 · d7f0ef2 · d7f0ef2
1 parent 1389d83
commit d7f0ef2
Show file tree

Hide file tree

Showing 5 changed files with 265 additions and 212 deletions.
diff --git a/GPU/Workflow/include/GPUWorkflow/GPUWorkflowSpec.h b/GPU/Workflow/include/GPUWorkflow/GPUWorkflowSpec.h
@@ -88,6 +88,7 @@ class GPURecoWorkflowSpec : public o2::framework::Task
   struct Config {
     int itsTriggerType = 0;
     int lumiScaleMode = 0;
+    int enableDoublePipeline = 0;
     bool decompressTPC = false;
     bool decompressTPCFromROOT = false;
     bool caClusterer = false;
@@ -149,8 +150,7 @@ class GPURecoWorkflowSpec : public o2::framework::Task
   int runITSTracking(o2::framework::ProcessingContext& pc);
 
   CompletionPolicyData* mPolicyData;
-  std::unique_ptr<o2::algorithm::ForwardParser<o2::tpc::ClusterGroupHeader>> mParser;
-  std::unique_ptr<GPUO2Interface> mTracker;
+  std::unique_ptr<GPUO2Interface> mGPUReco;
   std::unique_ptr<GPUDisplayFrontendInterface> mDisplayFrontend;
   std::unique_ptr<TPCFastTransform> mFastTransform;
   std::unique_ptr<TPCFastTransform> mFastTransformRef;

diff --git a/GPU/Workflow/src/GPUWorkflowITS.cxx b/GPU/Workflow/src/GPUWorkflowITS.cxx
@@ -308,7 +308,7 @@ void GPURecoWorkflowSpec::initFunctionITS(InitContext& ic)
   std::transform(mITSMode.begin(), mITSMode.end(), mITSMode.begin(), [](unsigned char c) { return std::tolower(c); });
   o2::its::VertexerTraits* vtxTraits = nullptr;
   o2::its::TrackerTraits* trkTraits = nullptr;
-  mTracker->GetITSTraits(trkTraits, vtxTraits, mITSTimeFrame);
+  mGPUReco->GetITSTraits(trkTraits, vtxTraits, mITSTimeFrame);
   mITSVertexer = std::make_unique<o2::its::Vertexer>(vtxTraits);
   mITSTracker = std::make_unique<o2::its::Tracker>(trkTraits);
   mITSVertexer->adoptTimeFrame(*mITSTimeFrame);