我是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)
...
我希望有人可以帮我解决这个问题。