PyCUDA中memcpy_htod()函数的用法及示例：实现主机内存到设备内存数据复制

在PyCUDA中，memcpy_htod()函数用于将主机内存中的数据复制到设备内存中。该函数的原型如下：

memcpy_htod(dest, src)

其中，dest指定了目标设备内存的地址，src指定了源主机内存的地址。

下面是一个示例，展示了如何使用memcpy_htod()函数将一个 NumPy 数组复制到设备内存中：

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

# 创建一个 1D 的 NumPy 数组
host_array = np.array([1, 2, 3, 4, 5], dtype=np.int32)

# 获取设备内存的指针
device_ptr = cuda.mem_alloc(host_array.nbytes)

# 将主机内存的数据复制到设备内存
cuda.memcpy_htod(device_ptr, host_array)

# 打印设备内存的数据
device_array = np.empty_like(host_array)
cuda.memcpy_dtoh(device_array, device_ptr)
print("Device Array:", device_array)

在上面的示例中，首先创建了一个包含五个整数的一维 NumPy 数组host_array。然后通过cuda.mem_alloc()函数获取了设备内存的指针device_ptr。最后使用cuda.memcpy_htod()函数将主机内存的数据复制到设备内存。

为了验证数据是否成功复制到设备内存，我们使用cuda.memcpy_dtoh()函数将设备内存的数据复制到一个新的 NumPy 数组device_array中，并打印它的值。

运行以上示例，输出将会是：

Device Array: [1 2 3 4 5]

这表明主机内存中的数据已经成功复制到设备内存中。