AthHIPExamples::DeviceStateHandler Class Reference

Base class to provide common infrastructure for handling multiprocess state transitions for GPU. More...

#include <DeviceStateHandler.h>

Inheritance diagram for AthHIPExamples::DeviceStateHandler:

Collaboration diagram for AthHIPExamples::DeviceStateHandler:

Public Member Functions
void	handle (const Incident &incident) override
virtual	~DeviceStateHandler ()=default

Protected Member Functions
virtual StatusCode	initialize_global ()
	Glocal initialization.
virtual StatusCode	initialize_worker ()
	Initialization per process.
virtual StatusCode	stop_worker ()
	clean up
virtual StatusCode	initialize ()
virtual StatusCode	stop ()

Protected Attributes
bool	m_is_child = false

Detailed Description

Base class to provide common infrastructure for handling multiprocess state transitions for GPU.

Based on CaloRecGPU/CaloGPUCUDAInitialization.h

Definition at line 27 of file DeviceStateHandler.h.

Constructor & Destructor Documentation

~DeviceStateHandler()

virtual AthHIPExamples::DeviceStateHandler::~DeviceStateHandler ( )

virtualdefault

Member Function Documentation

handle()

void AthHIPExamples::DeviceStateHandler::handle ( const Incident & incident )

inlineoverride

Definition at line 86 of file DeviceStateHandler.h.

  {
    const bool is_multiprocess = (Gaudi::Concurrency::ConcurrencyFlags::numProcs() > 0);
    if (is_multiprocess && incident.type() == AthenaInterprocess::UpdateAfterFork::type())
      {
        m_is_child = true;
        if (!this->initialize_worker().isSuccess())
        {
          throw GaudiException("Failed to initialize the device setup!",
                               "DeviceStateHandler::handle",
                               StatusCode::FAILURE);
        }
      }
  }

initialize()

virtual StatusCode AthHIPExamples::DeviceStateHandler::initialize ( )

inlineprotectedvirtual

Definition at line 49 of file DeviceStateHandler.h.

  {
    ATH_CHECK(this->initialize_global());
 
    const bool is_multiprocess = (Gaudi::Concurrency::ConcurrencyFlags::numProcs() > 0);
 
    if (is_multiprocess)
      {
        ServiceHandle<IIncidentSvc> incidentSvc("IncidentSvc","DeviceStateHandler");
        incidentSvc->addListener(this, AthenaInterprocess::UpdateAfterFork::type());
      }
    else
      {
        ATH_CHECK(this->initialize_worker());
      }
 
    return StatusCode::SUCCESS;
  }

initialize_global()

virtual StatusCode AthHIPExamples::DeviceStateHandler::initialize_global ( )

inlineprotectedvirtual

Glocal initialization.

Definition at line 32 of file DeviceStateHandler.h.

  {
    return StatusCode::SUCCESS;
  }

initialize_worker()

virtual StatusCode AthHIPExamples::DeviceStateHandler::initialize_worker ( )

inlineprotectedvirtual

Initialization per process.

Reimplemented in AthHIPExamples::LinearTransformExampleAlg.

Definition at line 38 of file DeviceStateHandler.h.

  {
    return StatusCode::SUCCESS;
  }

stop()

virtual StatusCode AthHIPExamples::DeviceStateHandler::stop ( )

inlineprotectedvirtual

Definition at line 68 of file DeviceStateHandler.h.

                            {
    const bool is_multiprocess = (Gaudi::Concurrency::ConcurrencyFlags::numProcs() > 0);
 
    if (!is_multiprocess) {
      ATH_CHECK(this->stop_worker());
    } else {
      if (m_is_child){
              ATH_CHECK(this->stop_worker());
      }
    }
 
  return StatusCode::SUCCESS;
  }

stop_worker()

virtual StatusCode AthHIPExamples::DeviceStateHandler::stop_worker ( )

inlineprotectedvirtual

clean up

Definition at line 44 of file DeviceStateHandler.h.

  {
    return StatusCode::SUCCESS;
  }

Member Data Documentation

m_is_child

bool AthHIPExamples::DeviceStateHandler::m_is_child = false

protected

Definition at line 82 of file DeviceStateHandler.h.

---

The documentation for this class was generated from the following file:

• DeviceStateHandler.h