cuda - 使用 CUDA 的矩阵乘法——错误的结果

Question

我有以下用于矩阵操作的内核代码。矩阵 A = 1*3 和矩阵 B = 3*3 结果矩阵 C 将是 1*3。在下面的代码中，宽度为 3。

__global__void MatrixMulKernel(float* d_M,float* d_N,float* d_P,int Width) {
   int row = blockIdx.y * blockDim.y + threadIdx.y;
   int col = blockIdx.x * blockDim.x + threadIdx.x;
    if(row>=Width || col>=Width){  // matrix range
      return;
    }
   float P_val = 0.0f;
   for (int k = 0; k < Width; ++k) {
   float M_elem = d_M[row * Width + k];
   float N_elem = d_N[k * Width + col];
   P_val += M_elem * N_elem;
  }
 d_p[row*Width+col] = P_val;
}

我内核代码调用如下

int block_size = 32;
dim3 dimGrid(Width/block_size, Width/block_size);
dim3 dimBlock(block_size, block size);
MatrixMulKernel<<<dimGrid, dimBlock>>>(d_M, d_N, d_P,3);

但我得到了错误的结果。我的结果总是为零。谁能帮助我。

Answer 1

该代码看起来像它用于乘以相同大小的 2 个方阵。

宽度是第一个矩阵的列数。

您必须将此作为参数提供给函数。