Public Member Functions
	CUDAMonitoringService (edm::ParameterSet const &iConfig, edm::ActivityRegistry &iRegistry)

void	postEvent (edm::StreamContext const &sc)

void	postModuleBeginStream (edm::StreamContext const &, edm::ModuleCallingContext const &mcc)

void	postModuleConstruction (edm::ModuleDescription const &desc)

	~CUDAMonitoringService ()=default

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

Private Attributes
int	numberOfDevices_ = 0

Detailed Description

Definition at line 21 of file CUDAMonitoringService.cc.

Constructor & Destructor Documentation

◆ CUDAMonitoringService()

CUDAMonitoringService::CUDAMonitoringService	(	edm::ParameterSet const &	iConfig,
		edm::ActivityRegistry &	iRegistry
	)

Definition at line 36 of file CUDAMonitoringService.cc.

                                                                                                          {
   // make sure that CUDA is initialised, and that the CUDAService destructor is called after this service's destructor
   edm::Service<CUDAService> cudaService;
   if (!cudaService->enabled())
     return;
   numberOfDevices_ = cudaService->numberOfDevices();
  
   if (config.getUntrackedParameter<bool>("memoryConstruction")) {
     registry.watchPostModuleConstruction(this, &CUDAMonitoringService::postModuleConstruction);
   }
   if (config.getUntrackedParameter<bool>("memoryBeginStream")) {
     registry.watchPostModuleBeginStream(this, &CUDAMonitoringService::postModuleBeginStream);
   }
   if (config.getUntrackedParameter<bool>("memoryPerEvent")) {
     registry.watchPostEvent(this, &CUDAMonitoringService::postEvent);
   }
 }

References CUDAService::enabled(), CUDAService::numberOfDevices(), numberOfDevices_, postEvent(), postModuleBeginStream(), postModuleConstruction(), edm::ActivityRegistry::watchPostEvent(), edm::ActivityRegistry::watchPostModuleBeginStream(), and edm::ActivityRegistry::watchPostModuleConstruction().

◆ ~CUDAMonitoringService()

CUDAMonitoringService::~CUDAMonitoringService ( )

default

Member Function Documentation

◆ fillDescriptions()

void CUDAMonitoringService::fillDescriptions ( edm::ConfigurationDescriptions & descriptions )

static

Definition at line 54 of file CUDAMonitoringService.cc.

                                                                                        {
   edm::ParameterSetDescription desc;
  
   desc.addUntracked<bool>("memoryConstruction", false)
       ->setComment("Print memory information for each device after the construction of each module");
   desc.addUntracked<bool>("memoryBeginStream", true)
       ->setComment("Print memory information for each device after the beginStream() of each module");
   desc.addUntracked<bool>("memoryPerEvent", true)
       ->setComment("Print memory information for each device after each event");
  
   descriptions.add("CUDAMonitoringService", desc);
   descriptions.setComment(
       "The memory information is the global state of the device. This gets confusing if there are multiple processes "
       "running on the same device. Probably the information retrieval should be re-thought?");
 }

References edm::ConfigurationDescriptions::add(), edm::ParameterSetDescription::addUntracked(), and edm::ConfigurationDescriptions::setComment().

◆ postEvent()

void CUDAMonitoringService::postEvent ( edm::StreamContext const & sc )

Definition at line 101 of file CUDAMonitoringService.cc.

                                                                 {
   auto log = edm::LogPrint("CUDAMonitoringService");
   log << "CUDA device memory after event";
   dumpUsedMemory(log, numberOfDevices_);
 }

References dqm-mbProfile::log, and numberOfDevices_.

Referenced by CUDAMonitoringService().

◆ postModuleBeginStream()

void CUDAMonitoringService::postModuleBeginStream	(	edm::StreamContext const &	,
		edm::ModuleCallingContext const &	mcc
	)

Definition at line 94 of file CUDAMonitoringService.cc.

                                                                                                              {
   auto log = edm::LogPrint("CUDAMonitoringService");
   log << "CUDA device memory after beginStream() of " << mcc.moduleDescription()->moduleLabel() << " ("
       << mcc.moduleDescription()->moduleName() << ")";
   dumpUsedMemory(log, numberOfDevices_);
 }

References dqm-mbProfile::log, edm::ModuleCallingContext::moduleDescription(), edm::ModuleDescription::moduleLabel(), edm::ModuleDescription::moduleName(), and numberOfDevices_.

Referenced by CUDAMonitoringService().

◆ postModuleConstruction()

void CUDAMonitoringService::postModuleConstruction ( edm::ModuleDescription const & desc )

Definition at line 88 of file CUDAMonitoringService.cc.

                                                                                    {
   auto log = edm::LogPrint("CUDAMonitoringService");
   log << "CUDA device memory after construction of " << desc.moduleLabel() << " (" << desc.moduleName() << ")";
   dumpUsedMemory(log, numberOfDevices_);
 }