StandardSequences/python/Services_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 from SimGeneral.HepPDTESSource.pythiapdt_cfi import *
0004
0005 # random numbers initialization service
0006 from IOMC.RandomEngine.IOMC_cff import *
0007
0008 # DQM store service
0009 from DQMServices.Core.DQMStore_cfi import *
0010
0011 # load ProcessAccelerators (that set the e.g. the necessary CUDA
0012 # stuff) when the "gpu" or "pixelNtupletFit" modifiers are enabled
0013 def _addProcessAccelerators(process):
0014     process.load("Configuration.StandardSequences.Accelerators_cff")
0015
0016 from Configuration.ProcessModifiers.gpu_cff import gpu
0017 from Configuration.ProcessModifiers.pixelNtupletFit_cff import pixelNtupletFit
0018 from Configuration.ProcessModifiers.alpaka_cff import alpaka
0019
0020 modifyConfigurationStandardSequencesServicesAddProcessAccelerators_ = (gpu | pixelNtupletFit | alpaka).makeProcessModifier(_addProcessAccelerators)
0021
0022 # load TritonService when SONIC workflow is enabled
0023 def _addTritonService(process):
0024     process.load("HeterogeneousCore.SonicTriton.TritonService_cff")
0025 from Configuration.ProcessModifiers.enableSonicTriton_cff import enableSonicTriton
0026 modifyConfigurationStandardSequencesServicesAddTritonService_ = enableSonicTriton.makeProcessModifier(_addTritonService)