| @@ -124,8 +124,9 @@ int MAIN__(int argc, char *argv[]){ | |||||
| FLOAT alpha[] = {1.0, 1.0}; | FLOAT alpha[] = {1.0, 1.0}; | ||||
| FLOAT beta [] = {1.0, 1.0}; | FLOAT beta [] = {1.0, 1.0}; | ||||
| char trans='N'; | char trans='N'; | ||||
| blasint m, i, j; | |||||
| blasint m, n, i, j; | |||||
| int loops = 1; | int loops = 1; | ||||
| int has_param_n=0; | |||||
| int l; | int l; | ||||
| char *p; | char *p; | ||||
| @@ -162,6 +163,11 @@ int MAIN__(int argc, char *argv[]){ | |||||
| if ( p != NULL ) | if ( p != NULL ) | ||||
| loops = atoi(p); | loops = atoi(p); | ||||
| if ((p = getenv("OPENBLAS_PARAM_N"))) { | |||||
| n = atoi(p); | |||||
| has_param_n=1; | |||||
| } | |||||
| #ifdef linux | #ifdef linux | ||||
| srandom(getpid()); | srandom(getpid()); | ||||
| @@ -174,7 +180,14 @@ int MAIN__(int argc, char *argv[]){ | |||||
| timeg=0; | 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++) | for (l=0; l<loops; l++) | ||||
| { | { | ||||
| @@ -189,7 +202,7 @@ int MAIN__(int argc, char *argv[]){ | |||||
| gettimeofday( &start, (struct timezone *)0); | 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); | gettimeofday( &stop, (struct timezone *)0); | ||||
| @@ -202,7 +215,7 @@ int MAIN__(int argc, char *argv[]){ | |||||
| timeg /= loops; | timeg /= loops; | ||||
| fprintf(stderr, | fprintf(stderr, | ||||
| " %10.2f MFlops\n", | " %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); | |||||
| } | } | ||||