SiPixelClusterizer/python/SiPixelClusterizerPreSplitting_cfi.py

0001 import FWCore.ParameterSet.Config as cms
0002 from HeterogeneousCore.CUDACore.SwitchProducerCUDA import SwitchProducerCUDA
0003
0004 # SiPixelGainCalibrationServiceParameters
0005 from CondTools.SiPixel.SiPixelGainCalibrationService_cfi import *
0006
0007 # legacy pixel cluster producer
0008 from RecoLocalTracker.SiPixelClusterizer.SiPixelClusterizer_cfi import siPixelClusters as _siPixelClusters
0009 siPixelClustersPreSplitting = SwitchProducerCUDA(
0010     cpu = _siPixelClusters.clone()
0011 )
0012
0013 from Configuration.Eras.Modifier_phase2_tracker_cff import phase2_tracker
0014 from Configuration.ProcessModifiers.gpu_cff import gpu
0015
0016 # ensure the same results when running on GPU (which supports only the 'HLT' payload) and CPU
0017 # but not for phase-2 where we don't calibrate digis in the clusterizer (yet)
0018 (gpu & ~phase2_tracker).toModify(siPixelClustersPreSplitting,
0019     cpu = dict(
0020         payloadType = 'HLT'
0021     )
0022 )
0023
0024 # SwitchProducer wrapping the legacy pixel cluster producer or an alias for the pixel clusters information converted from SoA
0025 gpu.toModify(siPixelClustersPreSplitting,
0026     cuda = cms.EDAlias(
0027         siPixelDigisClustersPreSplitting = cms.VPSet(
0028             cms.PSet(type = cms.string("SiPixelClusteredmNewDetSetVector"))
0029         )
0030     )
0031 )
0032
0033 from Configuration.ProcessModifiers.siPixelDigiMorphing_cff import siPixelDigiMorphing
0034 siPixelDigiMorphing.toModify(
0035     siPixelClustersPreSplitting,
0036     cpu = dict(
0037          src = 'siPixelDigisMorphed'
0038     )
0039 )