From 346a400027ca5579b3299999ee0c23615dcea1f4 Mon Sep 17 00:00:00 2001
From: Gino Lu <gino.lu@amd.com>
Date: Wed, 17 Sep 2025 04:19:44 -0500
Subject: [PATCH] fix time error

---
 example/ck_tile/18_flatmm/mxgemm/mx_flatmm.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/example/ck_tile/18_flatmm/mxgemm/mx_flatmm.cpp b/example/ck_tile/18_flatmm/mxgemm/mx_flatmm.cpp
index 54b016bc38..c358bf043d 100644
--- a/example/ck_tile/18_flatmm/mxgemm/mx_flatmm.cpp
+++ b/example/ck_tile/18_flatmm/mxgemm/mx_flatmm.cpp
@@ -199,10 +199,11 @@ float mx_flatmm_calc(const ck_tile::ScaleFlatmmHostArgs<ScaleM, ScaleN>& args,
             preprocess = clear_gemm_output;
         }
 
-        return ck_tile::launch_kernel_time_mask(
+        ave_time = ck_tile::launch_kernel_time_mask(
             s,
             preprocess,
             ck_tile::make_kernel<FlatmmConfig::kBlockPerCu>(Kernel{}, grids, blocks, 0, kargs));
+        return ave_time;
     };
 
     const auto RunSplitk = [&](const auto has_hot_loop_, const auto tail_number_) {