使用pycuda.drivermemcpy_htod()函数实现主机内存到设备内存的数据复制操作

在使用 PyCuda 进行 GPU 编程时，可以使用 pycuda.driver.memcpy_htod() 函数将数据从主机内存复制到设备内存。该函数的原型为：

memcpy_htod(dest, src)

dest 为目标设备内存地址，src 为源主机内存地址。它将数据从主机内存 src 复制到设备内存 dest。

下面是一个使用例子，展示了如何使用 pycuda.driver.memcpy_htod() 函数实现主机内存到设备内存的数据复制操作。

import pycuda.driver as drv
import pycuda.autoinit
import numpy as np

# 创建源数据并初始化
data_host = np.array([1, 2, 3, 4, 5], dtype=np.int32)

# 在设备上分配内存
data_device = drv.mem_alloc(data_host.nbytes)

# 将数据从主机内存复制到设备内存
drv.memcpy_htod(data_device, data_host)

# 打印设备上的数据
data_host_copy = np.empty_like(data_host)
drv.memcpy_dtoh(data_host_copy, data_device)
print("Device data:", data_host_copy)

在这个例子中，我们首先创建了一个长度为5的整数数组 data_host，并初始化为 [1, 2, 3, 4, 5]。然后，我们使用 drv.mem_alloc() 函数在设备上分配了与 data_host 相同大小的内存，并将其返回给 data_device。

接下来，我们调用 drv.memcpy_htod() 函数，将主机内存中的数据复制到设备内存。在这个例子中，我们通过 data_device 将数据复制到设备内存。

最后，我们使用 drv.memcpy_dtoh() 函数，将设备内存中的数据复制到主机内存中的 data_host_copy 数组，并打印出复制后的数据。在这个例子中，我们可以看到打印出的 data_host_copy 数组与初始的 data_host 数组相同，表明数据复制成功。

使用 pycuda.driver.memcpy_htod() 函数，可以方便地实现主机内存到设备内存的数据复制操作，为 GPU 编程提供了便利。