mt/bb_matmul.c

   1 #include "dataset.h"
   2 void __attribute__((noinline)) matmul(const int coreid, const int ncores, const int lda,  const data_t A[], const data_t B[], data_t C[] )
   3 {
   4   int i, j, k;
   5
   6   for (i = 0; i < lda; i += 2) {
   7     for (j = coreid * (lda / ncores); j < (coreid + 1) * (lda / ncores); j += 4) {
   8       register data_t c00 = 0, c01 = 0;
   9       register data_t c10 = 0, c11 = 0;
  10       register data_t c20 = 0, c21 = 0;
  11       register data_t c30 = 0, c31 = 0;
  12
  13       register data_t a0, a1, a2, a3, b0, b1;
  14       for (k = 0; k < lda; k++) {
  15         a0 = A[j*lda + k + 0*lda];
  16         a1 = A[j*lda + k + 1*lda];
  17         a2 = A[j*lda + k + 2*lda];
  18         a3 = A[j*lda + k + 3*lda];
  19
  20         b0 = B[k*lda + i + 0];
  21         b1 = B[k*lda + i + 1];
  22
  23         c00 += a0 * b0; c01 += a0 * b1;
  24         c10 += a1 * b0; c11 += a1 * b1;
  25         c20 += a2 * b0; c21 += a2 * b1;
  26         c30 += a3 * b0; c31 += a3 * b1;
  27       }
  28
  29       C[i + j*lda + 0 + 0*lda] = c00; C[i + j*lda + 1 + 0*lda] = c01;
  30       C[i + j*lda + 0 + 1*lda] = c10; C[i + j*lda + 1 + 1*lda] = c11;
  31       C[i + j*lda + 0 + 2*lda] = c20; C[i + j*lda + 1 + 2*lda] = c21;
  32       C[i + j*lda + 0 + 3*lda] = c30; C[i + j*lda + 1 + 3*lda] = c31;
  33     }
  34   }
  35 }