int j = blockIdx.y * blockDim.y + threadIdx.y;

//matrix multiplication

for (int k = 0; k < nk; k++)

sum += d_a[i * nk + k] * d_b[k * nj + j];