CUDA: невозможно вычислить размер сетки

Question

Я пишу программу для преобразования изображения rgba в оттенки серого. Я много работал над этим и правильно реализовал ядро. Тем не менее, размер сетки может быть неправильным, даже если он правильный по моей логике.

Ядро:

__global__ 
void rgba_to_greyscale(const uchar4* const rgbaImage, 
        unsigned char* const greyImage, 
        int numRows, int numCols) 
{ 
    int x = (blockIdx.x * blockDim.x) + threadIdx.x; 
    int y = (blockIdx.y * blockDim.y) + threadIdx.y; 

    if(x >= numCols || y >= numRows) 
     return; 

    uchar4 rgba = rgbaImage[x+y]; 
    float channelSum = 0.299f*rgba.x + 0.587f*rgba.y + 0.114f*rgba.z; 

    greyImage[x+y] = channelSum; 
} 

и запуск ядра:

const dim3 blockSize(10, 10, 1); //TODO 
    size_t gridSizeX, gridSizeY; 
    gridSizeX = numCols + (10 - (numCols % 10)); //adding some number to make it multiple of 10 
    gridSizeY = numRows + (10 - (numRows % 10)); //adding some number to make it multiple of 10 

    const dim3 gridSize(gridSizeX, gridSizeY, 1); //TODO 
    rgba_to_greyscale<<<gridSize, blockSize>>>(d_rgbaImage, d_greyImage, numRows, numCols); 

Я создаю больше числа потоков, то требуется, а затем применяя связанную проверку в ядре.

Answer 1

Вы получаете доступ к своему изображению, используя x+y. Но подумайте об этом, максимальный размер изображения, который вы можете получить таким образом, - numRows+numCols. Вы не можете просто добавить эти две координаты, поскольку это будет означать, что, например, (1,2) - тот же элемент изображения, что и (3,0), который является обычным мусором. Вместо этого для каждой координаты y вы должны пропустить всю строку изображения, поэтому она должна быть rgbaImage[x+y*numCols] (и то же самое для greyImage, конечно). Но обратите внимание, что в зависимости от компоновки ваших данных изображения это может быть и наоборот (x*numRows+y), но я предполагаю, что здесь используется обычный макет изображения (и в вашем ядре это не имеет значения, так как все пиксели обрабатываются одинаково).