RMM  23.12
RAPIDS Memory Manager
Classes | Functions
cuda_device.hpp File Reference
#include <rmm/detail/error.hpp>
#include <cuda_runtime_api.h>
Include dependency graph for cuda_device.hpp:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  rmm::cuda_device_id
 Strong type for a CUDA device identifier. More...
 
struct  rmm::cuda_set_device_raii
 RAII class that sets the current CUDA device to the specified device on construction and restores the previous device on destruction. More...
 

Functions

cuda_device_id rmm::get_current_cuda_device ()
 Returns a cuda_device_id for the current device. More...
 
int rmm::get_num_cuda_devices ()
 Returns the number of CUDA devices in the system. More...