GPU: pass pointer of services to doublePipieline run function

Author: ehellbar

GPU/Workflow/include/GPUWorkflow/GPUWorkflowSpec.h

@@ -186,12 +186,12 @@ class GPURecoWorkflowSpec : public o2::framework::Task
   template <class D, class E, class F, class G, class H, class I, class J, class K>
   void processInputs(o2::framework::ProcessingContext&, D&, E&, F&, G&, bool&, H&, I&, J&, K&);
 
-  int32_t runMain(o2::framework::ProcessingContext* pc, GPUTrackingInOutPointers* ptrs, GPUInterfaceOutputs* outputRegions, int32_t threadIndex = 0, GPUInterfaceInputUpdate* inputUpdateCallback = nullptr);
+  int32_t runMain(o2::framework::ProcessingContext* pc, ServiceRegistryRef* services, GPUTrackingInOutPointers* ptrs, GPUInterfaceOutputs* outputRegions, int32_t threadIndex = 0, GPUInterfaceInputUpdate* inputUpdateCallback = nullptr);
   int32_t runITSTracking(o2::framework::ProcessingContext& pc);
 
   int32_t handlePipeline(o2::framework::ProcessingContext& pc, GPUTrackingInOutPointers& ptrs, gpurecoworkflow_internals::GPURecoWorkflowSpec_TPCZSBuffers& tpcZSmeta, o2::gpu::GPUTrackingInOutZS& tpcZS, std::unique_ptr<gpurecoworkflow_internals::GPURecoWorkflow_QueueObject>& context);
   void RunReceiveThread();
-  void RunWorkerThread(int32_t id);
+  void RunWorkerThread(int32_t id, o2::framework::ServiceRegistryRef* services);
   void ExitPipeline();
   void handlePipelineEndOfStream(o2::framework::EndOfStreamContext& ec);
   void handlePipelineStop();

GPU/Workflow/src/GPUWorkflowPipeline.cxx

@@ -70,12 +70,12 @@ void GPURecoWorkflowSpec::initPipeline(o2::framework::InitContext& ic)
     };
     mPipeline->receiveThread = std::thread([this]() { RunReceiveThread(); });
     for (uint32_t i = 0; i < mPipeline->workers.size(); i++) {
-      mPipeline->workers[i].thread = std::thread([this, i]() { RunWorkerThread(i); });
+      mPipeline->workers[i].thread = std::thread([this, i]() { RunWorkerThread(i, &ic.services()); });
     }
   }
 }
 
-void GPURecoWorkflowSpec::RunWorkerThread(int32_t id)
+void GPURecoWorkflowSpec::RunWorkerThread(int32_t id, o2::framework::ServiceRegistryRef* services)
 {
   LOG(debug) << "Running pipeline worker " << id;
   auto& workerContext = mPipeline->workers[id];
@@ -91,7 +91,7 @@ void GPURecoWorkflowSpec::RunWorkerThread(int32_t id)
       workerContext.inputQueue.pop();
     }
     context->jobThreadIndex = id;
-    context->jobReturnValue = runMain(nullptr, context->jobPtrs, context->jobOutputRegions, id, context->jobInputUpdateCallback.get());
+    context->jobReturnValue = runMain(nullptr, services, context->jobPtrs, context->jobOutputRegions, id, context->jobInputUpdateCallback.get());
     {
       std::lock_guard lk(context->jobFinishedMutex);
       context->jobFinished = true;

GPU/Workflow/src/GPUWorkflowSpec.cxx

@@ -508,7 +508,7 @@ void GPURecoWorkflowSpec::processInputs(ProcessingContext& pc, D& tpcZSmeta, E&
   }
 }
 
-int32_t GPURecoWorkflowSpec::runMain(o2::framework::ProcessingContext* pc, GPUTrackingInOutPointers* ptrs, GPUInterfaceOutputs* outputRegions, int32_t threadIndex, GPUInterfaceInputUpdate* inputUpdateCallback)
+int32_t GPURecoWorkflowSpec::runMain(o2::framework::ProcessingContext* pc, o2::framework::ServiceRegistryRef* services, GPUTrackingInOutPointers* ptrs, GPUInterfaceOutputs* outputRegions, int32_t threadIndex, GPUInterfaceInputUpdate* inputUpdateCallback)
 {
   int32_t retVal = 0;
   if (mConfParam->dump < 2) {
@@ -520,8 +520,8 @@ int32_t GPURecoWorkflowSpec::runMain(o2::framework::ProcessingContext* pc, GPUTr
     static bool first = true;
     if (first) {
       first = false;
-      if (pc->services().get<const o2::framework::DeviceSpec>().inputTimesliceId == 0) { // TPC ConfigurableCarams are somewhat special, need to construct by hand
-        o2::conf::ConfigurableParam::write(o2::base::NameConf::getConfigOutputFileName(pc->services().get<const o2::framework::DeviceSpec>().name, "rec_tpc"), "GPU_rec_tpc,GPU_rec,GPU_proc_param,GPU_proc,GPU_global,trackTuneParams");
+      if (services()->get<const o2::framework::DeviceSpec>().inputTimesliceId == 0) { // TPC ConfigurableCarams are somewhat special, need to construct by hand
+        o2::conf::ConfigurableParam::write(o2::base::NameConf::getConfigOutputFileName(services()->get<const o2::framework::DeviceSpec>().name, "rec_tpc"), "GPU_rec_tpc,GPU_rec,GPU_proc_param,GPU_proc,GPU_global,trackTuneParams");
       }
     }
   }
@@ -843,7 +843,7 @@ void GPURecoWorkflowSpec::run(ProcessingContext& pc)
       mNextThreadIndex = (mNextThreadIndex + 1) % 2;
     }
 
-    retVal = runMain(&pc, &ptrs, &outputRegions, threadIndex);
+    retVal = runMain(&pc, nullptr, &ptrs, &outputRegions, threadIndex);
   }
   if (retVal != 0) {
     debugTFDump = true;