使用纹理1D的PyCuda 1D插值

2024-10-01 17:25:36 发布

您现在位置:Python中文网/ 问答频道 /正文

我是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)
...

我希望有人能帮我解决这个问题。你知道吗


Tags: 代码pos图像pycudaimgdriverout内核
1条回答
网友
1楼 · 发布于 2024-10-01 17:25:36

因为已经将线性内存绑定到纹理引用,所以必须在内核中使用^{},而不是^{}来访问纹理。你知道吗

注意,在这种情况下,也不可能执行插值,只能执行查找。你知道吗

相关问题 更多 >

    热门问题