#include <rmm/aligned.hpp>
#include <rmm/detail/error.hpp>
#include <rmm/detail/export.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... | |
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... | |