块矩阵乘法

Question

我想执行块矩阵乘法（将matirix分成多个sxs矩阵并乘以相应的块）。我已经按照Hennesy的体系结构书的示例代码编写了代码：

for(int jj=0;jj<=(n/s);jj += s){ 
      for(int kk=1;kk<=(n/s);kk += s){ 
        for(int i=1;i<=(n/s);i++){ 
          for(int j = jj; j<=((jj+s-1)>(n/s)?(n/s):(jj+s-1)); j++){ 
            temp = 0; 
            for(int k = kk; k<=((kk+s-1)>(n/s)?(n/s):(kk+s-1)); k++){ 
              temp += b[i][k]*a[k][j]; 
            } 
            c[j][i] += temp; 
          } 
        } 
      } 
    } 

这里，nxn是原始矩阵的大小。 a，b矩阵大小相同。我将a，b矩阵分成大小为sxs的块。在我的程序中，我已经给出了块大小为4.我把a，b的所有元素设置为5，常量和n = 1000。但是，我的结果中出现错误的值。我在这里做错了什么？从过去的2小时中吸取了这一点。如果可能的话，你们可以帮忙吗？在书中的参考码是这样的：

for (jj = 0; jj <= size; jj += N) { 
    for (kk = 1; kk <= size; kk += N) { 
     for (i = 1; i <= size; i++) { 
      for (j = jj; j <= findMin(jj+N-1, size); j++) { 
       temp = 0; 
       for (k = kk; k <= findMin(kk+N-1, size); k++) { 
        temp += B[i][k] * A[j][k]; 
       } 
       C[j][i] += temp; 
      } 
     } 
    } 
} 

在此，S = N和大小= N/S

Answer 1

for(int jj=0;jj<N;jj+= s){ 
     for(int kk=0;kk<N;kk+= s){ 
       for(int i=0;i<N;i++){ 
         for(int j = jj; j<((jj+s)>N?N:(jj+s)); j++){ 
           temp = 0; 
           for(int k = kk; k<((kk+s)>N?N:(kk+s)); k++){ 
             temp += a[i][k]*b[k][j]; 
           } 
           c[i][j] += temp; 
         } 
       } 
     } 
} 

AXB 大小是N

Answer 2

乍一看我很惊讶地看到0和1开始索引和< =用于循环终止测试。 Fortran或matlab代码的书籍有时会有1种索引，而c/C++使用0种索引。

您也可以分别实现和/或测试内部的两个for循环，因为它们将用于单块矩阵乘法。

我会保持findMin函数分开，而不是在循环测试中内联它。

Answer 3

错误在这一行。你有

temp += b[i][k]*a[k][j]; 

，你应该有

temp += b[i][k]*a[j][k]; 

代替。

这也将是更好，如果你可以把这块的功能，而不是这一行：

((jj+s-1)>(n/s)?(n/s):(jj+s-1));