EcalMonitorTasks/python/EcalMonitorTask_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 from DQM.EcalMonitorTasks.EcalMonitorTask_cfi import *
0004
0005 # Customization to run the CPU vs GPU comparison task if the job runs on a GPU enabled machine
0006 from Configuration.ProcessModifiers.gpuValidationEcal_cff import gpuValidationEcal
0007 from Configuration.ProcessModifiers.alpakaValidationEcal_cff import alpakaValidationEcal
0008 from DQM.EcalMonitorTasks.ecalGpuTask_cfi import ecalGpuTask
0009
0010 gpuValidationEcal.toModify(ecalGpuTask.params, runGpuTask = True)
0011 gpuValidationEcal.toModify(ecalMonitorTask.workers, func = lambda workers: workers.append("GpuTask"))
0012 gpuValidationEcal.toModify(ecalMonitorTask, workerParameters = dict(GpuTask = ecalGpuTask))
0013
0014 alpakaValidationEcal.toModify(ecalGpuTask.params, runGpuTask = True)
0015 alpakaValidationEcal.toModify(ecalMonitorTask.workers, func = lambda workers: workers.append("GpuTask"))
0016 alpakaValidationEcal.toModify(ecalMonitorTask, workerParameters = dict(GpuTask = ecalGpuTask))
0017
0018 # Skip consuming and running over the EcalRawData collection for all GPU WFs
0019 # This is to be used as long as the GPU unpacker unpacks a dummy EcalRawData collection
0020 from Configuration.ProcessModifiers.gpu_cff import gpu
0021 gpu.toModify(ecalMonitorTask.skipCollections, func = lambda skipCollections: skipCollections.append("EcalRawData"))
0022
0023 from Configuration.ProcessModifiers.alpaka_cff import alpaka
0024 alpaka.toModify(ecalMonitorTask.skipCollections, func = lambda skipCollections: skipCollections.append("EcalRawData"))