File indexing completed on 2023-10-25 10:02:14
0001 import FWCore.ParameterSet.Config as cms
0002
0003
0004
0005
0006 def customizePixelOnlyForProfilingGPUOnly(process):
0007
0008 process.consumer = cms.EDAnalyzer("GenericConsumer",
0009 eventProducts = cms.untracked.vstring('pixelTracksCUDA', 'pixelVerticesCUDA')
0010 )
0011
0012 process.consume_step = cms.EndPath(process.consumer)
0013
0014 process.schedule = cms.Schedule(process.raw2digi_step, process.reconstruction_step, process.consume_step)
0015
0016 return process
0017
0018
0019
0020
0021
0022
0023
0024
0025
0026 def customizePixelOnlyForProfilingGPUWithHostCopy(process):
0027
0028
0029
0030 process.consumer = cms.EDAnalyzer("GenericConsumer",
0031 eventProducts = cms.untracked.vstring('pixelTracksSoA', 'pixelVerticesSoA')
0032 )
0033
0034 process.consume_step = cms.EndPath(process.consumer)
0035
0036 process.schedule = cms.Schedule(process.raw2digi_step, process.reconstruction_step, process.consume_step)
0037
0038 return process
0039
0040
0041
0042
0043
0044
0045
0046
0047
0048
0049 def customizePixelOnlyForProfiling(process):
0050
0051 process.consumer = cms.EDAnalyzer("GenericConsumer",
0052 eventProducts = cms.untracked.vstring('pixelTracks', 'pixelVertices')
0053 )
0054
0055 process.consume_step = cms.EndPath(process.consumer)
0056
0057 process.schedule = cms.Schedule(process.raw2digi_step, process.reconstruction_step, process.consume_step)
0058
0059 return process