使用纹理1D进行PyCuda 1D插值

时间:2018-06-05 17:12:54

标签: python cuda textures pycuda

我是CUDA的新手,我的目标是使用PyCUDA和CUDA 1D纹理实现简单的1D插值。出于测试目的,我只想要一个内核,它返回一个数组中的原始图像值(从纹理中提取)。问题是tex1D(tex, pos);总是返回0。

这是我的CUDA内核代码:

interp1 = """

#include <stdint.h>

texture<uint8_t, 1> tex;


__global__ 
void interp1(uint8_t *out) {

    unsigned int pos = blockIdx.x * blockDim.x + threadIdx.x;
    out[pos] = tex1D(tex, pos);
}

"""

这是我的python代码剪切,我在测试图像中读取,在GPU上分配内存,将图像复制到GPU上,通过set_address创建我的纹理参考并调用我的内核:

...
img = cv2.imread("lena.jpg", 0)    
img_in = pycuda.driver.to_device(img.flatten())
texref.set_address(img_in, img.nbytes)
texref.set_format(pycuda.driver.array_format.UNSIGNED_INT8, 1)

img_out = pycuda.driver.mem_alloc(img.nbytes)

interp1_func(img_out, block=(512, 1, 1), grid=(7200, 1, 1)) # image is 1920 x 1920

context.synchronize()
imgnew = np.zeros_like(img.flatten())
pycuda.driver.memcpy_dtoh(imgnew, img_out)
imgnew = imgnew.reshape(img.shape)
...

我希望有人可以帮我解决这个问题。

1 个答案:

答案 0 :(得分:1)

因为您已将线性内存绑定到纹理引用,所以必须在内核中使用tex1Dfetch而不是tex1D来访问纹理。

请注意,在这种情况下,也无法执行插值,只能执行查找。