Move pixelTracksCPU module definition inside SwitchProducerCUDA

cms-sw · Apr 13, 2022 · d942c99 · d942c99
1 parent 532a37c
commit d942c99
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/RecoPixelVertexing/PixelTrackFitting/python/PixelTracks_cff.py b/RecoPixelVertexing/PixelTrackFitting/python/PixelTracks_cff.py
@@ -93,17 +93,14 @@
 
 from RecoPixelVertexing.PixelTriplets.pixelTracksCUDA_cfi import pixelTracksCUDA as _pixelTracksCUDA
 
-#Pixel tracks in SoA format on the CPU
-pixelTracksCPU = _pixelTracksCUDA.clone(
-    pixelRecHitSrc = "siPixelRecHitsPreSplitting",
-    idealConditions = False,
-    onGPU = False
-)
-
 # SwitchProducer providing the pixel tracks in SoA format on the CPU
 pixelTracksSoA = SwitchProducerCUDA(
     # build pixel ntuplets and pixel tracks in SoA format on the CPU
-    cpu = pixelTracksCPU
+    cpu = _pixelTracksCUDA.clone(
+        pixelRecHitSrc = "siPixelRecHitsPreSplitting",
+        idealConditions = False,
+        onGPU = False
+    )
 )
 # use quality cuts tuned for Run 2 ideal conditions for all Run 3 workflows
 run3_common.toModify(pixelTracksSoA.cpu,
@@ -132,7 +129,7 @@
 
 # "Patatrack" sequence running on GPU (or CPU if not available)
 from Configuration.ProcessModifiers.gpu_cff import gpu
-(pixelNtupletFit & gpu).toModify(pixelTracksCPU,
+(pixelNtupletFit & gpu).toModify(pixelTracksSoA.cpu,
     pixelRecHitSrc = "siPixelRecHitsPreSplittingSoA",
 )