#include <rmm/aligned.hpp>
#include <rmm/detail/error.hpp>
#include <rmm/detail/export.hpp>
#include <cuda_runtime_api.h>
#include <cstddef>
#include <utility>
类 | |
rmm::cuda_device_id | |
CUDA 设备标识符的强类型。更多... | |
rmm::cuda_set_device_raii | |
一个 RAII 类,在构造时将当前 CUDA 设备设置为指定的设备,并在析构时恢复之前的设备。更多... | |
函数 | |
cuda_device_id | rmm::get_current_cuda_device () |
返回当前设备的 cuda_device_id 。更多... | |
int | rmm::get_num_cuda_devices () |
返回系统中的 CUDA 设备数量。更多... | |
std::pair< std::size_t, std::size_t > | rmm::available_device_memory () |
返回当前设备的可用设备内存和总设备内存(以字节为单位)。更多... | |
std::size_t | rmm::percent_of_free_device_memory (int percent) |
返回当前 CUDA 设备上可用设备内存的大约指定百分比,并(向下)对齐到最接近的 CUDA 分配大小。更多... | |