Configuration/python/customizeHcalOnlyForProfiling.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 # Customise the HCAL-only reconstruction to run on GPU
0004 #
0005 # Currently, this means:
0006 #   - running the unpacker on CPU, converting the digis into SoA format and copying them to GPU;
0007 #   - running the HBHE local reconstruction, including MAHI, on GPU.
0008 def customizeHcalOnlyForProfilingGPUOnly(process):
0009
0010   process.consumer = cms.EDAnalyzer("GenericConsumer",
0011       eventProducts = cms.untracked.vstring('hbheRecHitProducerGPU')
0012   )
0013
0014   process.consume_step = cms.EndPath(process.consumer)
0015
0016   process.schedule = cms.Schedule(process.raw2digi_step, process.reconstruction_step, process.consume_step)
0017
0018   return process
0019
0020
0021 # Customise the HCAL-only reconstruction to run on GPU, and copy the data to the host
0022 #
0023 # Currently, this means:
0024 #   - running the unpacker on CPU, converting the digis into SoA format and copying them to GPU;
0025 #   - running the HBHE local reconstruction, including MAHI, on GPU;
0026 #   - copying the rechits to CPU and converting them to legacy format.
0027 #
0028 # (this is equivalent to customizeHcalOnlyForProfiling, as the copy and conversion is done by the same module)
0029 def customizeHcalOnlyForProfilingGPUWithHostCopy(process):
0030
0031   process.consumer = cms.EDAnalyzer("GenericConsumer",
0032       eventProducts = cms.untracked.vstring('hbheprereco')
0033   )
0034
0035   process.consume_step = cms.EndPath(process.consumer)
0036
0037   process.schedule = cms.Schedule(process.raw2digi_step, process.reconstruction_step, process.consume_step)
0038
0039   return process
0040
0041
0042 # Customise the HCAL-only reconstruction to run on GPU, copy the data to the host, and convert to legacy format
0043 #
0044 # Currently, this means:
0045 #   - running the unpacker on CPU, converting the digis into SoA format and copying them to GPU;
0046 #   - running the HBHE local reconstruction, including MAHI, on GPU;
0047 #   - copying the rechits to CPU and converting them to legacy format.
0048 #
0049 # The same customisation can be also used on the CPU workflow, running up to the rechits on CPU.
0050 def customizeHcalOnlyForProfiling(process):
0051
0052   process.consumer = cms.EDAnalyzer("GenericConsumer",
0053       eventProducts = cms.untracked.vstring('hbheprereco')
0054   )
0055
0056   process.consume_step = cms.EndPath(process.consumer)
0057
0058   process.schedule = cms.Schedule(process.raw2digi_step, process.reconstruction_step, process.consume_step)
0059
0060   return process