CUDAMonitoringService Class Reference

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)
void	postModuleEvent (edm::StreamContext const &sc, edm::ModuleCallingContext const &mcc)
	~CUDAMonitoringService ()=default

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

Private Attributes
int	numberOfDevices_ = 0

Detailed Description

Definition at line 22 of file CUDAMonitoringService.cc.

Constructor & Destructor Documentation

CUDAMonitoringService()

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

Definition at line 38 of file CUDAMonitoringService.cc.

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

                                                                                                         {
  // 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>("memoryPerModule")) {
    registry.watchPostModuleEvent(this, &CUDAMonitoringService::postModuleEvent);
  }
  if (config.getUntrackedParameter<bool>("memoryPerEvent")) {
    registry.watchPostEvent(this, &CUDAMonitoringService::postEvent);
  }
}

~CUDAMonitoringService()

CUDAMonitoringService::~CUDAMonitoringService ( )

default

Member Function Documentation

fillDescriptions()

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

static

Definition at line 59 of file CUDAMonitoringService.cc.

References edm::ConfigurationDescriptions::add(), submitPVResolutionJobs::desc, and edm::ConfigurationDescriptions::setComment().

                                                                                       {
  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>("memoryPerModule", true)
      ->setComment("Print memory information for each device after the event 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?");
}

postEvent()

void CUDAMonitoringService::postEvent

(

edm::StreamContext const &

sc

)

Definition at line 123 of file CUDAMonitoringService.cc.

References dqm-mbProfile::log, and numberOfDevices_.

Referenced by CUDAMonitoringService().

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

postModuleBeginStream()

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

Definition at line 109 of file CUDAMonitoringService.cc.

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

Referenced by CUDAMonitoringService().

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

postModuleConstruction()

void CUDAMonitoringService::postModuleConstruction

(

edm::ModuleDescription const &

desc

)

Definition at line 103 of file CUDAMonitoringService.cc.

References submitPVResolutionJobs::desc, dqm-mbProfile::log, and numberOfDevices_.

Referenced by CUDAMonitoringService().

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

postModuleEvent()

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

Definition at line 116 of file CUDAMonitoringService.cc.

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

Referenced by CUDAMonitoringService().

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

Member Data Documentation

numberOfDevices_

int CUDAMonitoringService::numberOfDevices_ = 0

private

Definition at line 35 of file CUDAMonitoringService.cc.

Referenced by CUDAMonitoringService(), postEvent(), postModuleBeginStream(), postModuleConstruction(), and postModuleEvent().