numpy - Pycuda 中 memcpy_htod 和 to_gpu 的区别？

Question

我正在学习 PyCUDA，在浏览 pycuda.gpuarray 的文档时，我对 pycuda.driver.memcpy_htod(也是 _dtoh)和 pycuda.gpuarray.to_gpu(也是 get)函数之间的区别感到困惑。根据 gpuarray documentation ， 。得到()。

例如，将 self 的内容转移到数组或新分配的 numpy.ndarray 中。如果给出数组，则它必须具有正确的大小(不一定是形状)和 dtype。如果未给出，则 pagelocked 指定新数组是否以页锁定方式分配。

这是说 .get() 的实现方式与 pycuda.driver.memcpy_dtoh 完全相同吗？不知何故，我想我误解了它。

Answer 1

pycuda.gpuarray.GPUArray.get()将 GPUArray 存储为 numpy.ndarray .

pycuda.driver.memcpy_dtoh()和 friend 在 CPU 和 GPU 内存之间复制普通缓冲区，而不对缓冲区中的数据进行任何处理。