Skip to content

Commit

Permalink
Load CUDAService from Services_cff, and only if gpu modifier is active (
Browse files Browse the repository at this point in the history
#432)

Include offlineBeamSpotCUDA only if gpu modifier is active
  • Loading branch information
makortel authored and fwyzard committed Jan 16, 2020
1 parent e41560b commit a68d9b8
Show file tree
Hide file tree
Showing 3 changed files with 11 additions and 7 deletions.
2 changes: 0 additions & 2 deletions Configuration/Applications/python/ConfigBuilder.py
Original file line number Diff line number Diff line change
Expand Up @@ -921,8 +921,6 @@ def define_Configs(self):
self.loadAndRemember('SimGeneral.HepPDTESSource.'+self._options.particleTable+'_cfi')

self.loadAndRemember('FWCore/MessageService/MessageLogger_cfi')
# Eventually replace with some more generic file to load
self.loadAndRemember('HeterogeneousCore/CUDAServices/CUDAService_cfi')

self.ALCADefaultCFF="Configuration/StandardSequences/AlCaRecoStreams_cff"
self.GENDefaultCFF="Configuration/StandardSequences/Generator_cff"
Expand Down
5 changes: 5 additions & 0 deletions Configuration/StandardSequences/python/Services_cff.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,3 +11,8 @@
#an "intermediate layer" remains, just in case somebody is using it...
# from Configuration.StandardSequences.SimulationRandomNumberGeneratorSeeds_cff import *
from DQMServices.Core.DQMStore_cfg import *

from Configuration.ProcessModifiers.gpu_cff import gpu
def loadCUDAService(process):
process.load("HeterogeneousCore.CUDAServices.CUDAService_cfi")
modifyConfigurationStandardSequencesServicesCUDAService = gpu.makeProcessModifier(loadCUDAService)
11 changes: 6 additions & 5 deletions RecoVertex/BeamSpotProducer/python/BeamSpot_cff.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,9 +3,10 @@
from RecoVertex.BeamSpotProducer.BeamSpot_cfi import *
from RecoVertex.BeamSpotProducer.beamSpotToCUDA_cfi import beamSpotToCUDA as _beamSpotToCUDA

offlineBeamSpotCUDA = _beamSpotToCUDA.clone()
offlineBeamSpotTask = cms.Task(offlineBeamSpot)

offlineBeamSpotTask = cms.Task(
offlineBeamSpot,
offlineBeamSpotCUDA
)
from Configuration.ProcessModifiers.gpu_cff import gpu
offlineBeamSpotCUDA = _beamSpotToCUDA.clone()
_offlineBeamSpotTask_gpu = offlineBeamSpotTask.copy()
_offlineBeamSpotTask_gpu.add(offlineBeamSpotCUDA)
gpu.toReplaceWith(offlineBeamSpotTask, _offlineBeamSpotTask_gpu)

0 comments on commit a68d9b8

Please sign in to comment.