CaloGPUCUDAInitialization Class Reference

Base class to provide some basic common infrastructure for initializing CUDA only at the right place to work fine with multiprocessing... More...

#include <CaloGPUCUDAInitialization.h>

Inheritance diagram for CaloGPUCUDAInitialization:

Collaboration diagram for CaloGPUCUDAInitialization:

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

Protected Member Functions
virtual StatusCode	initialize_non_CUDA ()
	Initialization that does not invoke CUDA functions. More...
virtual StatusCode	initialize_CUDA ()
	Initialization that invokes CUDA functions. More...
virtual StatusCode	initialize ()

Detailed Description

Base class to provide some basic common infrastructure for initializing CUDA only at the right place to work fine with multiprocessing...

Author

Nuno Fernandes nuno..nosp@m.dos..nosp@m.santo.nosp@m.s.fe.nosp@m.rnand.nosp@m.es@c.nosp@m.ern.c.nosp@m.h

Date

15 March 2024

Definition at line 27 of file CaloGPUCUDAInitialization.h.

Constructor & Destructor Documentation

~CaloGPUCUDAInitialization()

virtual CaloGPUCUDAInitialization::~CaloGPUCUDAInitialization ( )

virtualdefault

Member Function Documentation

handle()

void CaloGPUCUDAInitialization::handle ( const Incident &  incident )

inlineoverride

Definition at line 66 of file CaloGPUCUDAInitialization.h.

  {
    const bool is_multiprocess = (Gaudi::Concurrency::ConcurrencyFlags::numProcs() > 0);
    if (is_multiprocess && incident.type() == AthenaInterprocess::UpdateAfterFork::type())
      {
        if (!this->initialize_CUDA().isSuccess())
        {
          throw GaudiException("Failed to perform the CUDA initialization!",
                               "CaloGPUCUDAInitialization::handle",
                               StatusCode::FAILURE);
        }
      }
  }

initialize()

virtual StatusCode CaloGPUCUDAInitialization::initialize ( )

inlineprotectedvirtual

Reimplemented in GPUKernelSizeOptimizerSvc, CaloGPUHybridClusterProcessor, TopoAutomatonClustering, BasicGPUClusterInfoCalculator, GPUClusterInfoAndMomentsCalculator, and TopoAutomatonSplitting.

Definition at line 44 of file CaloGPUCUDAInitialization.h.

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

initialize_CUDA()

virtual StatusCode CaloGPUCUDAInitialization::initialize_CUDA ( )

inlineprotectedvirtual

Initialization that invokes CUDA functions.

Reimplemented in GPUKernelSizeOptimizerSvc, CaloGPUHybridClusterProcessor, TopoAutomatonClustering, BasicGPUClusterInfoCalculator, GPUClusterInfoAndMomentsCalculator, and TopoAutomatonSplitting.

Definition at line 39 of file CaloGPUCUDAInitialization.h.

  {
    return StatusCode::SUCCESS;
  }

initialize_non_CUDA()

virtual StatusCode CaloGPUCUDAInitialization::initialize_non_CUDA ( )

inlineprotectedvirtual

Initialization that does not invoke CUDA functions.

Reimplemented in CaloGPUHybridClusterProcessor, TopoAutomatonClustering, BasicGPUClusterInfoCalculator, GPUClusterInfoAndMomentsCalculator, and TopoAutomatonSplitting.

Definition at line 33 of file CaloGPUCUDAInitialization.h.

  {
    return StatusCode::SUCCESS;
  }

---

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

• CaloGPUCUDAInitialization.h