| 函数
cuda_device.hpp 文件参考
#include <rmm/aligned.hpp>
#include <rmm/detail/error.hpp>
#include <rmm/detail/export.hpp>
#include <cuda_runtime_api.h>
#include <cstddef>
#include <utility>
cuda_device.hpp 的包含依赖图
此图显示了哪些文件直接或间接包含此文件

转到此文件的源代码。

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