Inheritance diagram for ProcessAcceleratorCUDA.ProcessAcceleratorCUDA:

Public Member Functions
def	__init__ (self)

def	apply (self, process, accelerators)

def	enabledLabels (self)

def	labels (self)

Private Attributes
	_label

Detailed Description

Definition at line 5 of file ProcessAcceleratorCUDA.py.

Constructor & Destructor Documentation

◆ init()

def ProcessAcceleratorCUDA.ProcessAcceleratorCUDA.__init__ ( self )

Definition at line 6 of file ProcessAcceleratorCUDA.py.

     def __init__(self):
         super(ProcessAcceleratorCUDA,self).__init__()
         self._label = "gpu-nvidia"

Member Function Documentation

◆ apply()

def ProcessAcceleratorCUDA.ProcessAcceleratorCUDA.apply	(	self,
		process,
		accelerators
	)

Definition at line 17 of file ProcessAcceleratorCUDA.py.

References ProcessAcceleratorCUDA.ProcessAcceleratorCUDA._label, and LHECOMWeightProducer._label.

Referenced by heavyIonTools.ConfigureHeavyIons.__call__(), coreTools.RunOnData.__call__(), trackTools.MakeAODTrackCandidates.__call__(), runJetUncertainties.RunJetUncertainties.__call__(), metTools.AddMETCollection.__call__(), heavyIonTools.ProductionDefaults.__call__(), cmsswVersionTools.PickRelValInputFiles.__call__(), coreTools.RemoveMCMatching.__call__(), trackTools.MakePATTrackCandidates.__call__(), trigTools.SwitchOnTrigger.__call__(), runMETCorrectionsAndUncertainties.RunMETCorrectionsAndUncertainties.__call__(), heavyIonTools.SelectionDefaults.__call__(), heavyIonTools.DisbaleMonteCarloDeps.__call__(), trigTools.SwitchOnTriggerStandAlone.__call__(), tauTools.AddTauCollection.__call__(), trackTools.MakeTrackCandidates.__call__(), trigTools.SwitchOnTriggerMatching.__call__(), trigTools.SwitchOnTriggerMatchingStandAlone.__call__(), trigTools.SwitchOnTriggerMatchEmbedding.__call__(), jetTools.AddJetCollection.__call__(), jetTools.SwitchJetCollection.__call__(), jetTools.UpdateJetCollection.__call__(), jetTools.AddJetID.__call__(), and jetTools.SetTagInfos.__call__().

     def apply(self, process, accelerators):
         if not hasattr(process, "CUDAService"):
             from HeterogeneousCore.CUDAServices.CUDAService_cfi import CUDAService
             process.add_(CUDAService)
 
         if not hasattr(process.MessageLogger, "CUDAService"):
             process.MessageLogger.CUDAService = cms.untracked.PSet()
 
         if self._label in accelerators:
             process.CUDAService.enabled = True
         else:
             process.CUDAService.enabled = False
             
 cms.specialImportRegistry.registerSpecialImportForType(ProcessAcceleratorCUDA, "from HeterogeneousCore.CUDACore.ProcessAcceleratorCUDA import ProcessAcceleratorCUDA")

◆ enabledLabels()

def ProcessAcceleratorCUDA.ProcessAcceleratorCUDA.enabledLabels ( self )

Definition at line 11 of file ProcessAcceleratorCUDA.py.

References ProcessAcceleratorCUDA.ProcessAcceleratorCUDA.labels(), CollectionCombiner< Collection >.labels, confdb.HLTProcess.labels, trigger::HLTPrescaleTable.labels(), ecaldqm::binning::AxisSpecs.labels, Config.ProcessAccelerator.labels(), Config.ProcessAcceleratorTest.labels(), Config.ProcessAcceleratorTest2.labels(), and svgfig.Ticks.labels.

     def enabledLabels(self):
         enabled = (os.system("cudaIsEnabled") == 0)
         if enabled:
             return self.labels()
         else:
             return []