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 7 of file ProcessAcceleratorCUDA.py.

Constructor & Destructor Documentation

◆ init()

def ProcessAcceleratorCUDA.ProcessAcceleratorCUDA.__init__ ( self )

Definition at line 8 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 22 of file ProcessAcceleratorCUDA.py.

References ProcessAcceleratorCUDA.ProcessAcceleratorCUDA._label, LHECOMWeightProducer._label, and conddb2hdf5.DBDataProduct._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 self._label in accelerators:
             # Ensure that the CUDAService is loaded
             if not hasattr(process, "CUDAService"):
                 from HeterogeneousCore.CUDAServices.CUDAService_cfi import CUDAService
                 process.add_(CUDAService)
 
             # Propagate the CUDAService messages through the MessageLogger
             if not hasattr(process.MessageLogger, "CUDAService"):
                 process.MessageLogger.CUDAService = cms.untracked.PSet()
 
         else:
             # Make sure the CUDAService is not loaded
             if hasattr(process, "CUDAService"):
                 del process.CUDAService
 
             # Drop the CUDAService messages from the MessageLogger
             if hasattr(process.MessageLogger, "CUDAService"):
                 del process.MessageLogger.CUDAService
 
 
 # Ensure this module is kept in the configuration when dumping it
 cms.specialImportRegistry.registerSpecialImportForType(ProcessAcceleratorCUDA, "from HeterogeneousCore.CUDACore.ProcessAcceleratorCUDA import ProcessAcceleratorCUDA")

◆ enabledLabels()

def ProcessAcceleratorCUDA.ProcessAcceleratorCUDA.enabledLabels ( self )

Definition at line 15 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):
         # Check if CUDA is available, and if the system has at least one usable device.
         # These should be checked on each worker node, because it depends both
         # on the architecture and on the actual hardware present in the machine.
         status = PlatformStatus(os.waitstatus_to_exitcode(os.system("cudaIsEnabled")))
         return self.labels() if status == PlatformStatus.Success else []
 

◆ labels()

def ProcessAcceleratorCUDA.ProcessAcceleratorCUDA.labels ( self )

Definition at line 12 of file ProcessAcceleratorCUDA.py.

References ProcessAcceleratorCUDA.ProcessAcceleratorCUDA._label, LHECOMWeightProducer._label, and conddb2hdf5.DBDataProduct._label.

Referenced by ProcessAcceleratorCUDA.ProcessAcceleratorCUDA.enabledLabels(), and ProcessAcceleratorROCm.ProcessAcceleratorROCm.enabledLabels().

     def labels(self):
         return [ self._label ]
 

Member Data Documentation

◆ _label

ProcessAcceleratorCUDA.ProcessAcceleratorCUDA._label

private

Definition at line 10 of file ProcessAcceleratorCUDA.py.

Referenced by ProcessAcceleratorCUDA.ProcessAcceleratorCUDA.apply(), ProcessAcceleratorROCm.ProcessAcceleratorROCm.apply(), cmsswVersionTools.PickRelValInputFiles.apply(), validation.SimpleSample.digest(), ConfigToolBase.ConfigToolBase.dumpPython(), validation.SimpleSample.label(), ProcessAcceleratorCUDA.ProcessAcceleratorCUDA.labels(), ProcessAcceleratorROCm.ProcessAcceleratorROCm.labels(), cmsswVersionTools.PickRelValInputFiles.messageEmptyList(), trigTools.SwitchOnTrigger.toolCode(), trigTools.SwitchOnTriggerStandAlone.toolCode(), trigTools.SwitchOnTriggerMatching.toolCode(), trigTools.SwitchOnTriggerMatchingStandAlone.toolCode(), and trigTools.SwitchOnTriggerMatchEmbedding.toolCode().

Public Member Functions

Private Attributes