#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 分配大小。更多... | |