All Classes Files Functions Variables Typedefs Enumerations Enumerator Friends Modules Pages
Classes | Functions
cuda_device.hpp File Reference
#include <rmm/aligned.hpp>
#include <rmm/detail/error.hpp>
#include <rmm/detail/export.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...
 
std::pair< std::size_t, std::size_t > rmm::available_device_memory ()
 Returns the available and total device memory in bytes for the current device. More...
 
std::size_t rmm::percent_of_free_device_memory (int percent)
 Returns the approximate specified percent of available device memory on the current CUDA device, aligned (down) to the nearest CUDA allocation size. More...