AthCUDA::GPUSystemInfoSvc Class Reference

#include <GPUSystemInfoSvc.h>

Inheritance diagram for AthCUDA::GPUSystemInfoSvc:

Collaboration diagram for AthCUDA::GPUSystemInfoSvc:

Public Member Functions
virtual StatusCode	initialize () override
virtual void	handle (const Incident &) override
virtual const std::vector< DeviceInfo > &	getAvailableDevices () const override

Private Member Functions
void	readAvailableDevices ()

Private Attributes
std::vector< DeviceInfo >	m_deviceInfo
bool	m_wasChecked {}
std::once_flag	m_readDevicesOnceFlag

Detailed Description

Definition at line 18 of file GPUSystemInfoSvc.h.

Member Function Documentation

getAvailableDevices()

const std::vector< DeviceInfo > & AthCUDA::GPUSystemInfoSvc::getAvailableDevices ( ) const

overridevirtual

Definition at line 35 of file GPUSystemInfoSvc.cxx.

                                                                             {
        if (m_wasChecked == false) {
            ATH_MSG_WARNING("Function was called before service initialization!");
        }
        return m_deviceInfo;
    }

handle()

void AthCUDA::GPUSystemInfoSvc::handle ( const Incident & )

overridevirtual

Definition at line 30 of file GPUSystemInfoSvc.cxx.

                                                 {  
        readAvailableDevices();
        ATH_MSG_DEBUG("GPU availability check in UpdateAfterFork " << std::boolalpha << (m_deviceInfo.size() > 0));
    }

initialize()

StatusCode AthCUDA::GPUSystemInfoSvc::initialize ( )

overridevirtual

Definition at line 13 of file GPUSystemInfoSvc.cxx.

                                            {
        const bool isMultiprocess = (Gaudi::Concurrency::ConcurrencyFlags::numProcs() > 0);
        if (isMultiprocess) {
            SmartIF<IIncidentSvc> incsvc{service("IncidentSvc")};
            incsvc->addListener( this, AthenaInterprocess::UpdateAfterFork::type(), 1000); // high priority- before other GPU functions
 
            ATH_MSG_DEBUG("In multiprocess mode the CUDA context will be only checked after fork");
            return StatusCode::SUCCESS;
        }
        else {
            readAvailableDevices();
            
            ATH_MSG_DEBUG("GPU availability check in initialize " << std::boolalpha << (m_deviceInfo.size() > 0));
            return StatusCode::SUCCESS;
        }        
    }

readAvailableDevices()

void AthCUDA::GPUSystemInfoSvc::readAvailableDevices ( )

private

Definition at line 42 of file GPUSystemInfoSvc.cxx.

                                                {
        std::call_once(m_readDevicesOnceFlag, [this]() {
            std::string cudaErrorStr;
 
            int deviceCount = 0;
            cudaError_t error = cudaGetDeviceCount(&deviceCount);
            if (error != cudaSuccess) {
                cudaErrorStr = cudaGetErrorString(error);
                ATH_MSG_DEBUG("Error in cudaGetDeviceCount  " << cudaErrorStr);
            }
 
            for (int i = 0; i < deviceCount; i++) {
                cudaDeviceProp prop;
                error = cudaGetDeviceProperties(&prop, i);
 
                if (error != cudaSuccess) {
                    cudaErrorStr = cudaGetErrorString(error);
                    ATH_MSG_DEBUG("Error in cudaGetDeviceProperties for device " << i << " " << cudaErrorStr);
                    continue;
                }
                m_deviceInfo.emplace_back(i, prop.name, prop.major, prop.minor);
            }
 
 
            if (m_deviceInfo.empty()) {
                ATH_MSG_DEBUG("No CUDA Devices not available");
                return;
            }
 
            m_wasChecked = true;
        });
    }

Member Data Documentation

m_deviceInfo

std::vector<DeviceInfo> AthCUDA::GPUSystemInfoSvc::m_deviceInfo

private

Definition at line 27 of file GPUSystemInfoSvc.h.

m_readDevicesOnceFlag

std::once_flag AthCUDA::GPUSystemInfoSvc::m_readDevicesOnceFlag

private

Definition at line 29 of file GPUSystemInfoSvc.h.

m_wasChecked

bool AthCUDA::GPUSystemInfoSvc::m_wasChecked {}

private

Definition at line 28 of file GPUSystemInfoSvc.h.

{};

---

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

• GPUSystemInfoSvc.h
• GPUSystemInfoSvc.cxx