added code for the size of n

diff --git a/benchmark/gemm.c b/benchmark/gemm.c
index 4f9a588..347cf0d 100644 (file)
--- a/benchmark/gemm.c
+++ b/benchmark/gemm.c
@@ -124,8 +124,9 @@ int MAIN__(int argc, char *argv[]){
   FLOAT alpha[] = {1.0, 1.0};
   FLOAT beta [] = {1.0, 1.0};
   char trans='N';
-  blasint m, i, j;
+  blasint m, n, i, j;
   int loops = 1;
+  int has_param_n=0;
   int l;
   char *p;
 
@@ -162,6 +163,11 @@ int MAIN__(int argc, char *argv[]){
   if ( p != NULL )
        loops = atoi(p);
 
+  if ((p = getenv("OPENBLAS_PARAM_N"))) {
+          n = atoi(p);
+         has_param_n=1;          
+  }
+
 
 #ifdef linux
   srandom(getpid());
@@ -174,7 +180,14 @@ int MAIN__(int argc, char *argv[]){
 
     timeg=0;
 
-    fprintf(stderr, " %6d : ", (int)m);
+    if ( has_param_n == 1 && n <= m )
+       n=n;
+    else
+       n=m;
+
+
+
+    fprintf(stderr, " %6dx%d : ", (int)m, (int)n);
 
     for (l=0; l<loops; l++)
     {
@@ -189,7 +202,7 @@ int MAIN__(int argc, char *argv[]){
 
        gettimeofday( &start, (struct timezone *)0);
 
-       GEMM (&trans, &trans, &m, &m, &m, alpha, a, &m, b, &m, beta, c, &m );
+       GEMM (&trans, &trans, &m, &n, &m, alpha, a, &m, b, &m, beta, c, &m );
 
        gettimeofday( &stop, (struct timezone *)0);
 
@@ -202,7 +215,7 @@ int MAIN__(int argc, char *argv[]){
     timeg /= loops;
     fprintf(stderr,
            " %10.2f MFlops\n",
-           COMPSIZE * COMPSIZE * 2. * (double)m * (double)m * (double)m / timeg * 1.e-6);
+           COMPSIZE * COMPSIZE * 2. * (double)m * (double)m * (double)n / timeg * 1.e-6);
 
   }