我正在使用尺寸为3x3的水平掩模和尺寸为3x3的垂直掩模执行sobel边缘检测。 我首先在不使用共享内存的情况下实现了算法,现在我尝试使用共享内存来实现它。不使用共享内存时的性能优于何时使用。请有人解释一下。
代码
__global__ void image(int * in, int *out, int width){
int Mx[3][3] = {{-1,0,1},{-2,0,2},{-1,0,1}};
int My[3][3] = {{1,2,1},{0,0,0},{-1,-2,-1}};
__shared__ int sIn[t][t];
int sumX = 0, sumY = 0;
int row = blockIdx.y * blockDim.y + threadIdx.y;
int col = blockIdx.x * blockDim.x + threadIdx.x;
int r = threadIdx.y;
int c = threadIdx.x;
sIn[r][c] = in[(row*width) + col];
__syncthreads();
if(row == 0 || row == width-1 || col == 0 || col == width-1){
out[row*width + col] = 0;
}else{
for(int i = -1; i < 2; i++){
for(int j = -1; j < 2; j++){
int pixel;
if(r == 0 || r == (t-1) || c == 0 || c == (t-1))
pixel = in[(row + i) * width + (col + j)];
else
pixel = sIn[r + i][c + j];
sumX += pixel * Mx[i+1][j+1];
sumY += pixel * My[i+1][j+1];
}
}
int ans = abs(sumX) + abs(sumY);
if(ans > 255) ans = 255;
if(ans < 0) ans = 0;
out[row*width + col] = ans;
}
}
t是图块大小。
内核通话:
image<<<dim3(width/t, width/t, 1),dim3(t, t, 1)>>>(d_input, d_output, width);
答案 0 :(得分:-1)
我使用了围裙像素,从而增加了拼贴宽度,以便在检查3x3蒙版的边界像素时减少全局内存访问