summaryrefslogtreecommitdiff
path: root/test/routines/levelx
diff options
context:
space:
mode:
Diffstat (limited to 'test/routines/levelx')
-rw-r--r--test/routines/levelx/xconvgemm.hpp4
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);