File indexing completed on 2022-05-04 02:53:13
0001 import FWCore.ParameterSet.Config as cms
0002 from HeterogeneousCore.CUDACore.SwitchProducerCUDA import SwitchProducerCUDA
0003
0004 from RecoPixelVertexing.PixelTrackFitting.PixelTracks_cff import *
0005 from RecoPixelVertexing.PixelVertexFinding.PixelVertexes_cff import *
0006
0007
0008 pixelVerticesTask = cms.Task(
0009 pixelVertices
0010 )
0011
0012
0013 from Configuration.ProcessModifiers.pixelNtupletFit_cff import pixelNtupletFit
0014 from Configuration.Eras.Modifier_phase2_tracker_cff import phase2_tracker
0015
0016
0017 from RecoPixelVertexing.PixelVertexFinding.pixelVerticesCUDA_cfi import pixelVerticesCUDA as _pixelVerticesCUDA
0018 pixelVerticesSoA = SwitchProducerCUDA(
0019 cpu = _pixelVerticesCUDA.clone(
0020 pixelTrackSrc = "pixelTracksSoA",
0021 onGPU = False
0022 )
0023 )
0024
0025
0026 from RecoPixelVertexing.PixelVertexFinding.pixelVertexFromSoA_cfi import pixelVertexFromSoA as _pixelVertexFromSoA
0027 (pixelNtupletFit & ~phase2_tracker).toReplaceWith(pixelVertices, _pixelVertexFromSoA.clone(
0028 src = "pixelVerticesSoA"
0029 ))
0030
0031 (pixelNtupletFit & ~phase2_tracker).toReplaceWith(pixelVerticesTask, cms.Task(
0032
0033 pixelVerticesSoA,
0034
0035 pixelVertices
0036 ))
0037
0038
0039
0040 from Configuration.ProcessModifiers.gpu_cff import gpu
0041
0042
0043 pixelVerticesCUDA = _pixelVerticesCUDA.clone(
0044 pixelTrackSrc = "pixelTracksCUDA",
0045 onGPU = True
0046 )
0047
0048
0049 from RecoPixelVertexing.PixelVertexFinding.pixelVerticesSoA_cfi import pixelVerticesSoA as _pixelVerticesSoA
0050 gpu.toModify(pixelVerticesSoA,
0051 cuda = _pixelVerticesSoA.clone(
0052 src = cms.InputTag("pixelVerticesCUDA")
0053 )
0054 )
0055
0056
0057
0058 from Configuration.ProcessModifiers.gpuValidationPixel_cff import gpuValidationPixel
0059 (pixelNtupletFit & gpu & gpuValidationPixel).toModify(pixelVerticesSoA.cpu,
0060 pixelTrackSrc = "pixelTracksSoA@cpu"
0061 )
0062
0063 (pixelNtupletFit & gpu).toReplaceWith(pixelVerticesTask, cms.Task(
0064
0065 pixelVerticesCUDA,
0066
0067 pixelVerticesTask.copy()
0068 ))
0069
0070
0071 recopixelvertexingTask = cms.Task(
0072 pixelTracksTask,
0073 pixelVerticesTask
0074 )
0075 recopixelvertexing = cms.Sequence(recopixelvertexingTask)