Inheritance diagram for ProcessAcceleratorROCm.ProcessAcceleratorROCm:

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 ProcessAcceleratorROCm.py.

Constructor & Destructor Documentation

◆ init()

def ProcessAcceleratorROCm.ProcessAcceleratorROCm.__init__ ( self )

Definition at line 8 of file ProcessAcceleratorROCm.py.

     def __init__(self):
         super(ProcessAcceleratorROCm, self).__init__()
         self._label = "gpu-amd"
 

Member Function Documentation

◆ apply()

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

Definition at line 22 of file ProcessAcceleratorROCm.py.

References ProcessAcceleratorCUDA.ProcessAcceleratorCUDA._label, ProcessAcceleratorROCm.ProcessAcceleratorROCm._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 ROCmService is loaded
             if not hasattr(process, "ROCmService"):
                 from HeterogeneousCore.ROCmServices.ROCmService_cfi import ROCmService
                 process.add_(ROCmService)
 
             # Propagate the ROCmService messages through the MessageLogger
             if not hasattr(process.MessageLogger, "ROCmService"):
                 process.MessageLogger.ROCmService = cms.untracked.PSet()
 
         else:
             # Make sure the ROCmService is not loaded
             if hasattr(process, "ROCmService"):
                 del process.ROCmService
 
             # Drop the ROCmService messages from the MessageLogger
             if hasattr(process.MessageLogger, "ROCmService"):
                 del process.MessageLogger.ROCmService
 
 
 # Ensure this module is kept in the configuration when dumping it
 cms.specialImportRegistry.registerSpecialImportForType(ProcessAcceleratorROCm, "from HeterogeneousCore.ROCmCore.ProcessAcceleratorROCm import ProcessAcceleratorROCm")

◆ enabledLabels()

def ProcessAcceleratorROCm.ProcessAcceleratorROCm.enabledLabels ( self )

Definition at line 15 of file ProcessAcceleratorROCm.py.

References ProcessAcceleratorCUDA.ProcessAcceleratorCUDA.labels(), ProcessAcceleratorROCm.ProcessAcceleratorROCm.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 ROCm 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("rocmIsEnabled")))
         return self.labels() if status == PlatformStatus.Success else []
 

◆ labels()

def ProcessAcceleratorROCm.ProcessAcceleratorROCm.labels ( self )

Definition at line 12 of file ProcessAcceleratorROCm.py.

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

Referenced by ProcessAcceleratorROCm.ProcessAcceleratorROCm.enabledLabels().

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

Member Data Documentation

◆ _label

ProcessAcceleratorROCm.ProcessAcceleratorROCm._label

private

Definition at line 10 of file ProcessAcceleratorROCm.py.

Referenced by ProcessAcceleratorROCm.ProcessAcceleratorROCm.apply(), cmsswVersionTools.PickRelValInputFiles.apply(), validation.SimpleSample.digest(), ConfigToolBase.ConfigToolBase.dumpPython(), validation.SimpleSample.label(), 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