首页 > 解决方案 > 我想将 Thrust 与 CUDA 混合使用

问题描述

我有几个 CUDA 内核的 C++ 代码。使用 Thrust 库可以更轻松地执行某些操作。可以将推力调用与 CUDA 内核混合使用吗?

更具体地说,假设我有一个这样声明的推力装置矢量:

thrust::device_vector<int> d_myVector;

然后假设我使用推力对该设备向量执行一些操作。我以后可以直接从 CUDA 内核访问该设备向量吗?或者,我是否必须将推力设备向量复制回主机,然后将其转换为 CUDA 内核可以使用的东西,然后再将其复制回设备?

标签: cudathrust

解决方案


推荐阅读