Speed up benchmarks that take a long time to run

[riscv-tests.git] / benchmarks / mt-matmul / matmul.c

diff --git a/benchmarks/mt-matmul/matmul.c b/benchmarks/mt-matmul/matmul.c
index 7c23309aab654a7f539a56a0bd6a30f174b2c43b..6d7b97863a5a023ffffa2f04e4152b736a900784 100644 (file)
--- a/benchmarks/mt-matmul/matmul.c
+++ b/benchmarks/mt-matmul/matmul.c
@@ -2,21 +2,22 @@
 
 #include "dataset.h"
 #include "util.h"
 
 #include "dataset.h"
 #include "util.h"

+#include <stddef.h>

 
 

-void __attribute__((noinline)) matmul(const int coreid, const int ncores, const int lda,  const data_t A[], const data_t B[], data_t C[] )
+#pragma GCC optimize ("unroll-loops")
+
+void matmul(const size_t coreid, const size_t ncores, const size_t lda,  const data_t A[], const data_t B[], data_t C[])

 {
 {

-   int i, j, k;
-   int block = lda / ncores;
-   int start = block * coreid;
+  size_t i, j, k;
+  size_t block = lda / ncores;
+  size_t start = block * coreid;

  
  

-   for ( j = start; j < (start+block); j++ ) 
-   {
-      for ( k = 0; k < lda; k++ )  
-      {
-         for ( i = 0; i < lda; i++ ) 
-         {
-            C[i + j*lda] += A[j*lda + k] * B[k*lda + i];
-         }
-      }
-   }
+  for (i = 0; i < lda; i++) {
+    for (j = start; j < (start+block); j++) {
+      data_t sum = 0;
+      for (k = 0; k < lda; k++)
+        sum += A[j*lda + k] * B[k*lda + i];
+      C[i + j*lda] = sum;
+    }
+  }

 }
 }