c++ - OpenCL 回写缓冲区

Question

当我尝试在主机缓冲区上写回 opencl 代码的输出时，我总是得到一个零/空数组

这是内核:

void kernel convolution(global const int* A, global const int* B,
        global int* C, const int size_1, const int size_2) {
    float prob_1[256];
    float prob_2[256];
    float cum_prob_1[256];
    float cum_prob_2[256];
    prob_1[get_global_id(0)] = (A[get_global_id(0)]/size_1);
    prob_2[get_global_id(0)] = (B[get_global_id(0)]/size_2);
    cum_prob_1[0] = prob_1[0];
    cum_prob_1[get_global_id(0)+1] = cum_prob_1[get_global_id(0)] + prob_1[get_global_id(0)+1];
    cum_prob_2[0] = prob_2[0];
    cum_prob_2[get_global_id(0)+1] = cum_prob_2[get_global_id(0)] + prob_2[get_global_id(0)+1];
    int j = 0;
    for (int i = 0; i < 256; i++) {
        if (cum_prob_1[i] <= cum_prob_2[j]) {
            C[i] = j;
        }
        else {
            while (cum_prob_1[i] > cum_prob_2[j])
                j++;
            if (cum_prob_2[j] - cum_prob_1[i] > cum_prob_1[i] - cum_prob_2[j - 1]) {
                C[i] = j - 1;
            }
            else {
                C[i] = j;
            }
        }
    }
}

下面是剩下的代码:

// create buffers on the device
cl::Buffer buffer_A(context, CL_MEM_READ_WRITE, sizeof(int) * 256);
cl::Buffer buffer_B(context, CL_MEM_READ_WRITE, sizeof(int) * 256);
cl::Buffer buffer_C(context, CL_MEM_READ_WRITE, sizeof(int) * 256);

//create queue to which we will push commands for the device.
cl::CommandQueue queue(context, default_device);

//write arrays A and B to the device
queue.enqueueWriteBuffer(buffer_A, CL_TRUE, 0, sizeof(int) * 256, pixls_1);
queue.enqueueWriteBuffer(buffer_B, CL_TRUE, 0, sizeof(int) * 256, pixls_2);

//alternative way to run the kernel
cl::Kernel kernel_add = cl::Kernel(program, "convolution");
kernel_add.setArg(0, buffer_A);
kernel_add.setArg(1, buffer_B);
kernel_add.setArg(2, buffer_C);
kernel_add.setArg(3, size_1);
kernel_add.setArg(4, size_2);

queue.enqueueNDRangeKernel(kernel_add, cl::NullRange, cl::NDRange(sizeof(cl_mem)), cl::NullRange);
queue.finish();

int lookup_table[256];
for (size_t i = 0; i < 256; i++) {
    lookup_table[i] = 10;
}
//read result C from the device to array C
queue.enqueueReadBuffer(buffer_C, CL_TRUE, 0, sizeof(int) * 256, lookup_table);

//map the new pixel values from the lookup table
for (int i = 0; i < inp_image.total(); i++) {
    myData[i]= lookup_table[myData[i]];
}

system("pause");
return 0;

lookup_table 是给我零/空值的表，但它应该在所有单元格中给我 5...

Answer 1

我在您的代码中看到两件对我来说非常可疑的事情:

1. 调用 enqueueNDRangeKernel()，特别是使用 cl::NDRange(sizeof(cl_mem)) 指定全局工作项维度数组... AFAICT，这没有任何意义，但我很可能会错过这里的一些东西。

2. 在内核本身中使用临时数组:在我看来，您很可能忘记将它们声明为 local。会是这样吗？如果是这样，您需要在初始化部分之后添加一个barrier(CLK_LOCAL_MEM_FENCE)。

无论如何，我的感觉是您的代码旨在与大小为 256 的工作组一起使用，并为临时数组提供 local 内存。我可能是错的，但无论如何，内核的运行方式及其逻辑都存在问题。