diff options
Diffstat (limited to 'test/routines/level3/xgemm.hpp')
-rw-r--r-- | test/routines/level3/xgemm.hpp | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/test/routines/level3/xgemm.hpp b/test/routines/level3/xgemm.hpp index 4cfa9c83..7a473a4f 100644 --- a/test/routines/level3/xgemm.hpp +++ b/test/routines/level3/xgemm.hpp @@ -193,7 +193,13 @@ class TestXgemm { // Describes how to compute performance metrics static size_t GetFlops(const Arguments<T> &args) { - return 2 * args.m * args.n * args.k; + if((args.precision == Precision::kComplexSingle) || (args.precision == Precision::kComplexDouble)) { + // complex flops + return args.m * args.n * (8 * args.k - 2); + } else { + // scalar flops + return args.m * args.n * (2 * args.k - 1); + } } static size_t GetBytes(const Arguments<T> &args) { return (args.m*args.k + args.k*args.n + 2*args.m*args.n) * sizeof(T); |