python - PyCuda 中的 cudaBindTextureToArray

Question

有没有一种方法可以使用 PyCuda 将已经在 gpu 上的数组绑定(bind)到纹理？

已经有一个cuda.bind_array_to_texref(cuda.make_multichannel_2d_array(...), texref)将 CPU 上的数组绑定(bind)到纹理，但我找不到相当于 cudaBindTextureToArray 的等价物如果数组已经在设备上，则在 PyCuda 中。例如，做:

myArray = [1, 2, 3]
myArray_d = gpu.to_gpu(myArray)  # then performs some computations on  it, and then
cuda.bind_texture_to_array(myArray_d, texref)

Answer 1

如果您想将 GPU 内存中的现有 CUDA 数组绑定(bind)到纹理引用，那么 pycuda.driver.TextureReference.set_array() 可能就是您想要的。请注意 PyCUDA 是建立在驱动程序 API 上的，因此您要查找的调用实际上是 cuTexRefSetArray 而不是 cudaBindTextureToArray。