update bench_gemm

1 files changed, 11 insertions, 5 deletions

diff --git a/bench/bench_gemm.cpp b/bench/bench_gemm.cpp
index 8222271fb..0974ebe4c 100644
--- a/bench/bench_gemm.cpp
+++ b/bench/bench_gemm.cpp
@@ -148,7 +148,7 @@ int main(int argc, char ** argv)
   int m = s;
   int n = s;
   int p = s;
-  int cache_size = -1;
+  int cache_size1=-1, cache_size2=l2, cache_size3 = 0;
 
   bool need_help = false;
   for (int i=1; i<argc;)
@@ -169,7 +169,13 @@ int main(int argc, char ** argv)
       else if(argv[i][1]=='c')
       {
         ++i;
-        cache_size = atoi(argv[i++]);
+        cache_size1 = atoi(argv[i++]);
+        if(argv[i][0]!='-')
+        {
+          cache_size2 = atoi(argv[i++]);
+          if(argv[i][0]!='-')
+            cache_size3 = atoi(argv[i++]);
+        }
       }
       else if(argv[i][1]=='t')
       {
@@ -191,14 +197,14 @@ int main(int argc, char ** argv)
 
   if(need_help)
   {
-    std::cout << argv[0] << " -s <matrix sizes> -c <cache size> -t <nb tries> -p <nb repeats>\n";
+    std::cout << argv[0] << " -s <matrix sizes> -c <cache sizes> -t <nb tries> -p <nb repeats>\n";
     std::cout << "   <matrix sizes> : size\n";
     std::cout << "   <matrix sizes> : rows columns depth\n";
     return 1;
   }
 
-  if(cache_size>0)
-    setCpuCacheSizes(cache_size,96*cache_size);
+ if(cache_size1>0)
+   setCpuCacheSizes(cache_size1,cache_size2,cache_size3);
 
   
   A a(m,p); a.setRandom();