rmm::cuda_set_device_raii 结构体参考

一个 RAII 类，在构造时将当前 CUDA 设备设置为指定设备，并在析构时恢复先前的设备。 更多...

#include <cuda_device.hpp>

公共成员函数
	cuda_set_device_raii (cuda_device_id dev_id)
	构造一个新的 cuda_set_device_raii 对象，并将当前 CUDA 设备设置为 dev_id 更多...
	~cuda_set_device_raii () noexcept
	重新激活先前的 CUDA 设备。
	cuda_set_device_raii (cuda_set_device_raii const &)=delete
cuda_set_device_raii &	operator= (cuda_set_device_raii const &)=delete
	cuda_set_device_raii (cuda_set_device_raii &&)=delete
cuda_set_device_raii &	operator= (cuda_set_device_raii &&)=delete

详细描述

一个 RAII 类，在构造时将当前 CUDA 设备设置为指定设备，并在析构时恢复先前的设备。

构造函数与析构函数文档

cuda_set_device_raii()

rmm::cuda_set_device_raii::cuda_set_device_raii ( cuda_device_id  dev_id )

inlineexplicit

构造一个新的 cuda_set_device_raii 对象，并将当前 CUDA 设备设置为 dev_id

参数

dev_id

要设置为当前 CUDA 设备的设备

---

此结构体的文档生成自以下文件

• cuda_device.hpp