CUDA Device Management

Files
file	cuda_device.hpp

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...

Detailed Description

Function Documentation

get_current_cuda_device()

cuda_device_id rmm::get_current_cuda_device ( )

inline

Returns a cuda_device_id for the current device.

The current device is the device on which the calling thread executes device code.

Returns

cuda_device_id for the current device

get_num_cuda_devices()

int rmm::get_num_cuda_devices ( )

inline

Returns the number of CUDA devices in the system.

Returns

Number of CUDA devices in the system