diff options
author | Gael Guennebaud <g.gael@free.fr> | 2015-02-20 11:59:49 +0100 |
---|---|---|
committer | Gael Guennebaud <g.gael@free.fr> | 2015-02-20 11:59:49 +0100 |
commit | 333b4973833784370e823b732a389fdfbb0b217b (patch) | |
tree | 6adac2334326b8437d0654a4e4eadfb517c09dd5 | |
parent | 2da1594750de99304fecce96fbfd29d080736446 (diff) |
update bench_gemm
-rw-r--r-- | bench/bench_gemm.cpp | 16 |
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(); |