Lecture 2 – Threads - many-core.group

GPU strategy 2 – kernel (part 1)
// allocate an array in shared memory
__shared__ float temp[NI_TILE][NJ_TILE];
// find i and j indices of current thread
ti = threadIdx.x;
tj = threadIdx.y;
i = blockIdx.x*(NI_TILE-2) + ti;
j = blockIdx.y*(NJ_TILE-2) + tj;
// index into linear memory for current thread
i2d = i + ni*j;
// if thread is in domain, read from global to shared memory
if (i2d < ni*nj) {
temp[ti][tj] = temp_in[i2d];
}
// make sure all threads have read in data
__syncthreads();