编程拾遗

CUDA Thust库算法使用device路径,并使用原生指针作为参数

playerzhou 无回复

若原生指针ptr指向的内存区域为显卡内存,那么thrust库算法若使用ptr作为算法参数,需要使用device路径的函数
但thrust库无法识别ptr指向的是host内存还是device内存,因此会调用host路径的函数,造成访存失败。
因此需要使用device_ptr<T>对原生指针进行一层封装,这样之后thrust库就能知道该指针指向显存

来一发吐槽

此站点使用Akismet来减少垃圾评论。了解我们如何处理您的评论数据