Configuration/python/customizeEcalOnlyForProfiling.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 # Customise the ECAL-only reconstruction to run on GPU
0004 #
0005 # Currently, this means running only the unpacker and multifit, up to the uncalbrated rechits
0006 def customizeEcalOnlyForProfilingGPUOnly(process):
0007
0008   process.consumer = cms.EDAnalyzer("GenericConsumer",
0009       eventProducts = cms.untracked.vstring('ecalMultiFitUncalibRecHitGPU')
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 # Customise the ECAL-only reconstruction to run on GPU, and copy the data to the host
0020 #
0021 # Currently, this means running only the unpacker and multifit, up to the uncalbrated rechits
0022 def customizeEcalOnlyForProfilingGPUWithHostCopy(process):
0023
0024   process.consumer = cms.EDAnalyzer("GenericConsumer",
0025       eventProducts = cms.untracked.vstring('ecalMultiFitUncalibRecHitSoA')
0026   )
0027
0028   process.consume_step = cms.EndPath(process.consumer)
0029
0030   process.schedule = cms.Schedule(process.raw2digi_step, process.reconstruction_step, process.consume_step)
0031
0032   return process
0033
0034
0035 # Customise the ECAL-only reconstruction to run on GPU, copy the data to the host, and convert to legacy format
0036 #
0037 # Currently, this means running only the unpacker and multifit, up to the uncalbrated rechits, on the GPU
0038 # and the rechits producer on the CPU
0039 #
0040 # The same customisation can be also used on the CPU workflow, running up to the rechits on CPU.
0041 def customizeEcalOnlyForProfiling(process):
0042
0043   process.consumer = cms.EDAnalyzer("GenericConsumer",
0044       eventProducts = cms.untracked.vstring('ecalRecHit')
0045   )
0046
0047   process.consume_step = cms.EndPath(process.consumer)
0048
0049   process.schedule = cms.Schedule(process.raw2digi_step, process.reconstruction_step, process.consume_step)
0050
0051   return process