Fixed the performance client for convgemm and added GFLOPS measurements

1 files changed, 3 insertions, 1 deletions

diff --git a/test/routines/levelx/xconvgemm.hpp b/test/routines/levelx/xconvgemm.hpp
index 6ca5965b..7233f7b6 100644
--- a/test/routines/levelx/xconvgemm.hpp
+++ b/test/routines/levelx/xconvgemm.hpp
@@ -151,7 +151,9 @@ public:
 
   // Describes how to compute performance metrics
   static size_t GetFlops(const Arguments<T> &args) {
-    return args.batch_count; // TODO
+    const auto patch_size = args.kernel_h * args.kernel_w * args.channels;
+    const auto num_patches = OutputHeight(args) * OutputWidth(args);
+    return args.batch_count * 2 * num_patches * args.num_kernels * patch_size;
   }
   static size_t GetBytes(const Arguments<T> &args) {
     return (GetSizeA(args) + GetSizeB(args) + GetSizeC(args)) * sizeof(T);