gpt4 book ai didi

numpy - Pycuda 中 memcpy_htod 和 to_gpu 的区别?

转载 作者:行者123 更新时间:2023-12-05 01:06:34 24 4
gpt4 key购买 nike

我正在学习 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 完全相同吗?不知何故,我想我误解了它。

最佳答案

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

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

关于numpy - Pycuda 中 memcpy_htod 和 to_gpu 的区别?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/20034808/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com