Merge pull request cms-sw#15 from SegmentLinking/CMSSW_13_3_0_pre3_LS…

…T_X_LoadTwoSDLlibs Switch between CPU and CUDA backends without need to recompile the standalone
kmohrman · Oct 19, 2023 · f44eff7 · f44eff7
2 parents 2081692 + 2659399
commit f44eff7
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 20 deletions.
diff --git a/RecoTracker/IterativeTracking/python/HighPtTripletStep_cff.py b/RecoTracker/IterativeTracking/python/HighPtTripletStep_cff.py
@@ -387,14 +387,13 @@
 from RecoTracker.LST.lstSeedTracks_cfi import lstInitialStepSeedTracks,lstHighPtTripletStepSeedTracks
 from RecoTracker.LST.lstPixelSeedInputProducer_cfi import lstPixelSeedInputProducer
 from RecoTracker.LST.lstPhase2OTHitsInputProducer_cfi import lstPhase2OTHitsInputProducer
-from RecoTracker.LST.alpaka_cuda_asyncLSTProducer_cfi import alpaka_cuda_asyncLSTProducer
-from RecoTracker.LST.alpaka_serial_syncLSTProducer_cfi import alpaka_serial_syncLSTProducer
-from Configuration.ProcessModifiers.gpu_cff import gpu
-from HeterogeneousCore.CUDACore.SwitchProducerCUDA import SwitchProducerCUDA
-lstProducer = SwitchProducerCUDA(
-    cpu  = alpaka_serial_syncLSTProducer.clone(),
+from RecoTracker.LST.lstProducer_cfi import lstProducer as _lstProducer
+# Define lstProducer like this to be able to run the serial version explicitly
+lstProducer = _lstProducer.clone(
+  alpaka = cms.untracked.PSet(backend = cms.untracked.string('serial_sync'))
 )
-gpu.toModify(lstProducer,cuda = alpaka_cuda_asyncLSTProducer.clone())
+from Configuration.ProcessModifiers.gpu_cff import gpu
+gpu.toReplaceWith(lstProducer,_lstProducer.clone())
 _HighPtTripletStepTask_LST.add(siPhase2RecHits, lstInitialStepSeedTracks, lstHighPtTripletStepSeedTracks, lstPixelSeedInputProducer, lstPhase2OTHitsInputProducer, lstProducer)
 (trackingPhase2PU140 & trackingLST).toReplaceWith(HighPtTripletStepTask, _HighPtTripletStepTask_LST)
 

diff --git a/RecoTracker/LST/plugins/BuildFile.xml b/RecoTracker/LST/plugins/BuildFile.xml
@@ -22,15 +22,28 @@
 </library>
 
 <!-- alpaka-based portable plugins -->
-<library file="alpaka/*.cc" name="RecoTrackerLSTPluginsPortable">
+<library file="alpaka/*.cc" name="RecoTrackerLSTPluginsPortableCpu">
   <use name="alpaka"/>
-  <use name="lst"/>
+  <use name="lst_cpu"/>
   <use name="FWCore/Framework"/>
   <use name="FWCore/ParameterSet"/>
   <use name="FWCore/Utilities"/>
   <use name="HeterogeneousCore/AlpakaCore"/>
   <use name="HeterogeneousCore/AlpakaInterface"/>
   <use name="RecoTracker/LST"/>
-  <flags ALPAKA_BACKENDS="cuda serial"/>
+  <flags ALPAKA_BACKENDS="serial"/>
+  <flags EDM_PLUGIN="1"/>
+</library>
+
+<library file="alpaka/*.cc" name="RecoTrackerLSTPluginsPortableGpu">
+  <use name="alpaka"/>
+  <use name="lst_gpu"/>
+  <use name="FWCore/Framework"/>
+  <use name="FWCore/ParameterSet"/>
+  <use name="FWCore/Utilities"/>
+  <use name="HeterogeneousCore/AlpakaCore"/>
+  <use name="HeterogeneousCore/AlpakaInterface"/>
+  <use name="RecoTracker/LST"/>
+  <flags ALPAKA_BACKENDS="cuda"/>
   <flags EDM_PLUGIN="1"/>
 </library>
diff --git a/RecoTracker/LST/test/LSTAlpakaTester.py b/RecoTracker/LST/test/LSTAlpakaTester.py
@@ -90,16 +90,9 @@
 
 # Main LST Producer
 process.load('Configuration.StandardSequences.Accelerators_cff')
-process.load('RecoTracker.LST.alpaka_cuda_asyncLSTProducer_cfi')
-process.load('RecoTracker.LST.alpaka_serial_syncLSTProducer_cfi')
-### Enable alpaka-based heterogeneous modules
-process.AlpakaServiceCudaAsync = cms.Service('AlpakaServiceCudaAsync')
-process.AlpakaServiceSerialSync = cms.Service('AlpakaServiceSerialSync')
-from HeterogeneousCore.CUDACore.SwitchProducerCUDA import SwitchProducerCUDA
-process.lstProducer = SwitchProducerCUDA(
-        cpu = process.alpaka_serial_syncLSTProducer.clone(),
-        cuda = process.alpaka_cuda_asyncLSTProducer.clone()
-)
+process.load("HeterogeneousCore.AlpakaCore.ProcessAcceleratorAlpaka_cfi")
+from RecoTracker.LST.lstProducer_cfi import lstProducer as _lstProducer
+process.lstProducer = _lstProducer.clone()
 
 # Track Fitting
 import RecoTracker.TrackProducer.TrackProducer_cfi