RMM
23.12
RAPIDS Memory Manager
|
#include <rmm/detail/error.hpp>
#include <cuda_runtime_api.h>
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... | |