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.alpaka_cff import alpaka
0018
0019 modifyConfigurationStandardSequencesServicesAddProcessAccelerators_ = (gpu | alpaka).makeProcessModifier(_addProcessAccelerators)
0020
0021 # load TritonService when SONIC workflow is enabled
0022 def _addTritonService(process):
0023     process.load("HeterogeneousCore.SonicTriton.TritonService_cff")
0024 from Configuration.ProcessModifiers.enableSonicTriton_cff import enableSonicTriton
0025 modifyConfigurationStandardSequencesServicesAddTritonService_ = enableSonicTriton.makeProcessModifier(_addTritonService)