使用pycuda.drivermemcpy_htod()从主机内存拷贝数据到设备内存的示例代码

下面是使用 pycuda.driver.memcpy_htod() 函数从主机内存拷贝数据到设备内存的示例代码：

import pycuda.driver as cuda
import pycuda.autoinit

# 创建一个宿主端的数组
host_array = [1, 2, 3, 4, 5]

# 获取设备当前上下文
context = cuda.Context.get_current()

# 分配一个设备端的数组
device_array = cuda.mem_alloc(len(host_array) * 4)  # 4 字节大小

# 将数据从宿主端拷贝到设备端
cuda.memcpy_htod(device_array, host_array)

# 创建一个设备端的数组来接收数据
output_array = [0] * len(host_array)
output_device_array = cuda.mem_alloc(len(host_array) * 4)

# 拷贝数据从设备端到宿主端
cuda.memcpy_dtoh(output_array, output_device_array)

# 打印结果
print("Host array:", host_array)
print("Output array:", output_array)

这个例子中，我们首先创建了一个宿主端的数组 host_array，然后使用 cuda.mem_alloc() 分配了一个设备端的数组 device_array。接下来使用 cuda.memcpy_htod() 函数将数据从宿主端拷贝到设备端。

然后，我们创建了一个宿主端的数组 output_array 和一个设备端的数组 output_device_array，用于接收从设备端拷贝的结果。最后，我们使用 cuda.memcpy_dtoh() 函数将数据从设备端拷贝到宿主端，并打印结果。

这个例子中的数组都是 32 位整数，所以我们使用了 len(host_array) * 4 来分配相应的空间。如果你使用的是其他数据类型，需要适当调整为相应的字节大小。