nvidia_gpu.h File Reference

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Macros
#define	PAPI_NVML_DEV_BUFFER_SIZE   NVML_DEVICE_UUID_BUFFER_SIZE

Functions
void	open_nvidia_gpu_dev_type (_sysdetect_dev_type_info_t *dev_type_info)
void	close_nvidia_gpu_dev_type (_sysdetect_dev_type_info_t *dev_type_info)

Macro Definition Documentation

PAPI_NVML_DEV_BUFFER_SIZE

#define PAPI_NVML_DEV_BUFFER_SIZE   NVML_DEVICE_UUID_BUFFER_SIZE

Definition at line 7 of file nvidia_gpu.h.

Function Documentation

close_nvidia_gpu_dev_type()

void close_nvidia_gpu_dev_type

(

_sysdetect_dev_type_info_t *

dev_type_info

)

Definition at line 386 of file nvidia_gpu.c.

{
    papi_free(dev_type_info->dev_info_arr);
}

open_nvidia_gpu_dev_type()

void open_nvidia_gpu_dev_type

(

_sysdetect_dev_type_info_t *

dev_type_info

)

Definition at line 337 of file nvidia_gpu.c.

{
    memset(dev_type_info, 0, sizeof(*dev_type_info));
    dev_type_info->id = PAPI_DEV_TYPE_ID__CUDA;
    strcpy(dev_type_info->vendor, "NVIDIA");
    strcpy(dev_type_info->status, "Device Initialized");
 
#ifdef HAVE_CUDA
    if (load_cuda_sym(dev_type_info->status)) {
        return;
    }
 
    int dev, dev_count;
    CU_CALL((*cuDeviceGetCountPtr)(&dev_count), return);
    dev_type_info->num_devices = dev_count;
    if (dev_count == 0) {
        return;
    }
 
    _sysdetect_gpu_info_u *arr = papi_calloc(dev_count, sizeof(*arr));
    for (dev = 0; dev < dev_count; ++dev) {
        fill_dev_info(&arr[dev], dev);
    }
 
#ifdef HAVE_NVML
    if (!load_nvml_sym(dev_type_info->status)) {
        fill_dev_affinity_info(arr, dev_count);
        unload_nvml_sym();
    }
#else
    const char *message = "NVML not configured, no device affinity available";
    int count = snprintf(dev_type_info->status, PAPI_MAX_STR_LEN, "%s", message);
    if (count >= PAPI_MAX_STR_LEN) {
        SUBDBG("Status string truncated.");
    }
#endif /* HAVE_NVML */
 
    unload_cuda_sym();
    dev_type_info->dev_info_arr = (_sysdetect_dev_info_u *)arr;
#else
    const char *message = "CUDA not configured, no CUDA device available";
    int count = snprintf(dev_type_info->status, PAPI_MAX_STR_LEN, "%s", message);
    if (count >= PAPI_MAX_STR_LEN) {
        SUBDBG("Status string truncated.");
    }
#endif /* HAVE_CUDA */
}

Here is the call graph for this function: