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.
virtual StatusCode	initialize_CUDA ()
	Initialization that invokes CUDA functions.
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 BasicGPUClusterInfoCalculator, CaloGPUHybridClusterProcessor, GPUClusterInfoAndMomentsCalculator, GPUClusterSorter, GPUKernelSizeOptimizerSvc, TopoAutomatonClustering, 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;
  }