爲什麼我的CUDA內核不改變我的數組？

Question

我從頭開始自學CUDA。我製作了這個簡單的內核，爲2D數組中的每個相關元素添加1。 2D陣列的元素來自圖像的紅色通道（zebra.jpg）。

from pycuda.compiler import SourceModule 
import matplotlib.image as img 
import pycuda.driver as cuda 
import pycuda.autoinit 
import numpy as np 

my_image = img.imread("zebra.jpg")[:,:,0] 
block = (32, 32, 1) 
grid = (8, 8) 

if __name__ == '__main__': 
    width, height = np.int32(my_image.shape) 
    im = np.array(my_image) 
    print 'original sum: ' + str(np.sum(im)) 

    # Create the CUDA kernel, and run it. 
    mod = SourceModule(""" 
    __global__ void add1ToEverything(float* r, int w, int h) { 
     int rowID = blockDim.y * blockDim.y + threadIdx.y; 
     int colID = blockDim.x * blockIdx.x + threadIdx.x; 

     if (rowID > 0 && rowID < h - 2 && colID > 0 && colID < w - 2) 
     r[rowID * w + colID] += 1.0; 
    } 
    """) 
    func = mod.get_function('add1ToEverything') 

    for i in range(0, 5): 
    func(cuda.InOut(im), width, height, block=block, grid=grid) 
    print 'new sum: ' + str(np.sum(im)) 

不過，後來我運行這個程序，我得到下面的結果。

original sum: 1828815 
new sum: 1828815 

爲什麼我的原始金額與我的新金額相同？新的總和不應該更大？

這是zebra.jpg。

Answer 1

你的問題是在該行：
int rowID = blockDim.y * blockDim.y + threadIdx.y;
它應該是：
int rowID = blockDim.y * blockIdx.y + threadIdx.y;