Changed ESProducer to use assigned queue

SegmentLinking · Apr 22, 2024 · 354c6d3 · 354c6d3
1 parent 8e297ff
commit 354c6d3
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/RecoTracker/LST/plugins/alpaka/LSTModulesDevESProducer.cc b/RecoTracker/LST/plugins/alpaka/LSTModulesDevESProducer.cc
@@ -2,6 +2,7 @@
 
 #include "HeterogeneousCore/AlpakaCore/interface/alpaka/ESProducer.h"
 #include "HeterogeneousCore/AlpakaCore/interface/alpaka/ModuleFactory.h"
+#include "HeterogeneousCore/AlpakaCore/interface/alpaka/typelookup.h"
 #include "HeterogeneousCore/AlpakaInterface/interface/config.h"
 #include "HeterogeneousCore/AlpakaInterface/interface/memory.h"
 
@@ -29,7 +30,7 @@ class LSTModulesDevESProducer : public ESProducer {
 
   static void fillDescriptions(edm::ConfigurationDescriptions &descriptions);
 
-  std::optional<SDL::modulesBuffer<alpaka_common::DevHost>> produce(const TrackerRecoGeometryRecord &iRecord);
+  std::optional<SDL::modulesBuffer<alpaka_common::DevHost>> produce(device::Record<TrackerRecoGeometryRecord> const& iRecord);
 
 };
 
@@ -45,13 +46,14 @@ void LSTModulesDevESProducer::fillDescriptions(edm::ConfigurationDescriptions &d
   descriptions.addWithDefaultLabel(desc);
 }
 
-  std::optional<SDL::modulesBuffer<alpaka_common::DevHost>> LSTModulesDevESProducer::produce(const TrackerRecoGeometryRecord &iRecord) {
+  std::optional<SDL::modulesBuffer<alpaka_common::DevHost>> LSTModulesDevESProducer::produce(device::Record<TrackerRecoGeometryRecord> const& iRecord) {
     SDL::modulesBuffer<alpaka_common::DevHost> modules(cms::alpakatools::host());
-    alpaka::QueueCpuBlocking queue(cms::alpakatools::host());
+    SDL::QueueAcc& queue = iRecord.queue();
     SDL::LST<SDL::Acc>::loadAndFillES(queue, &modules);
     return modules;
 }
 
 }  // namespace ALPAKA_ACCELERATOR_NAMESPACE
 
 DEFINE_FWK_EVENTSETUP_ALPAKA_MODULE(LSTModulesDevESProducer);
+TYPELOOKUP_DATA_REG(ALPAKA_ACCELERATOR_NAMESPACE::ESDeviceProduct<SDL::modulesBuffer<alpaka_common::DevHost>>);