-
Notifications
You must be signed in to change notification settings - Fork 4.3k
/
Services_cff.py
26 lines (19 loc) · 1.15 KB
/
Services_cff.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import FWCore.ParameterSet.Config as cms
from SimGeneral.HepPDTESSource.pythiapdt_cfi import *
# random numbers initialization service
from IOMC.RandomEngine.IOMC_cff import *
# DQM store service
from DQMServices.Core.DQMStore_cfi import *
# load ProcessAccelerators (that set the e.g. the necessary CUDA
# stuff) when the "gpu" or "pixelNtupletFit" modifiers are enabled
def _addProcessAccelerators(process):
process.load("Configuration.StandardSequences.Accelerators_cff")
from Configuration.ProcessModifiers.gpu_cff import gpu
from Configuration.ProcessModifiers.pixelNtupletFit_cff import pixelNtupletFit
from Configuration.ProcessModifiers.alpaka_cff import alpaka
modifyConfigurationStandardSequencesServicesAddProcessAccelerators_ = (gpu | pixelNtupletFit | alpaka).makeProcessModifier(_addProcessAccelerators)
# load TritonService when SONIC workflow is enabled
def _addTritonService(process):
process.load("HeterogeneousCore.SonicTriton.TritonService_cff")
from Configuration.ProcessModifiers.enableSonicTriton_cff import enableSonicTriton
modifyConfigurationStandardSequencesServicesAddTritonService_ = enableSonicTriton.makeProcessModifier(_addTritonService)