diff options
Diffstat (limited to 'test/routines/levelx/xconvgemm.hpp')
-rw-r--r-- | test/routines/levelx/xconvgemm.hpp | 4 |
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); |