Configuration/StandardSequences/python/Services_cff.py - master - githubmirror/cmssw

Configuration/StandardSequences/python/Services_cff.py

25 строк1 KB

AdrianoDee

Removing CUDA/gpu from Pixel code configs, removing all CUDA wfs

07 янв 2025, 17:54

07 янв 2025, 17:54399e4f2

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.alpaka_cff import alpaka

modifyConfigurationStandardSequencesServicesAddProcessAccelerators_ = (gpu | 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)