Merge pull request #416 from JishinMaster/master

set the correct flop count for xgemm

1 files changed, 7 insertions, 1 deletions

diff --git a/test/routines/level3/xtrmm.hpp b/test/routines/level3/xtrmm.hpp
index 51377a16..f39db5cd 100644
--- a/test/routines/level3/xtrmm.hpp
+++ b/test/routines/level3/xtrmm.hpp
@@ -162,7 +162,13 @@ class TestXtrmm {
   // Describes how to compute performance metrics
   static size_t GetFlops(const Arguments<T> &args) {
     auto k = (args.side == Side::kLeft) ? args.m : args.n;
-    return args.m * args.n * k;
+    if((args.precision == Precision::kComplexSingle) || (args.precision == Precision::kComplexDouble)) {
+      // complex flops
+      return 4 * args.m * args.n * k;
+    } else {
+      // scalar flops
+      return args.m * args.n * k;
+    }
   }
   static size_t GetBytes(const Arguments<T> &args) {
     auto k = (args.side == Side::kLeft) ? args.m : args.n;