From 9a56ea579b8eb70977a66f0839cdf63dfc3a2654 Mon Sep 17 00:00:00 2001
From: yadaish <yadai@amd.com>
Date: Sun, 7 Dec 2025 15:29:14 +0000
Subject: [PATCH] update

---
 include/ck_tile/ops/flatmm/kernel/moe_flatmm_kernel.hpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/include/ck_tile/ops/flatmm/kernel/moe_flatmm_kernel.hpp b/include/ck_tile/ops/flatmm/kernel/moe_flatmm_kernel.hpp
index 62a69a1667..eba35ce318 100644
--- a/include/ck_tile/ops/flatmm/kernel/moe_flatmm_kernel.hpp
+++ b/include/ck_tile/ops/flatmm/kernel/moe_flatmm_kernel.hpp
@@ -601,7 +601,8 @@ struct MoeFlatmmKernel
         const auto& b_flat_tensor_view = [&]() {
             return make_naive_tensor_view<address_space_enum::global>(
                 b_flat_ptr,
-                make_tuple(kFlatN - kargs.n_padded_zeros / NPerXdl, kFlatK),
+                // make_tuple(kFlatN - kargs.n_padded_zeros / NPerXdl, kFlatK),
+                make_tuple(kFlatN, kFlatK),
                 make_tuple(kFlatK, 1),
                 number<FlatmmPipeline::GetVectorSizeB()>{},
                 number<1>{});