From c0adc147a34affe8dbf465c7a05175d511a5e411 Mon Sep 17 00:00:00 2001
From: Yi DING <yi.ding@amd.com>
Date: Wed, 26 Nov 2025 10:11:43 +0800
Subject: [PATCH] [CK_TILE] Fix Compilation of Flatmm Examples (#3285)

[ROCm/composable_kernel commit: c7dce2ac29136939b6fe6aabadd026e53dcf35c9]
---
 .../flatmm_pipeline_agmem_bgmem_creg_v1_policy.hpp     | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)
 mode change 100755 => 100644 include/ck_tile/ops/flatmm/pipeline/flatmm_pipeline_agmem_bgmem_creg_v1_policy.hpp

diff --git a/include/ck_tile/ops/flatmm/pipeline/flatmm_pipeline_agmem_bgmem_creg_v1_policy.hpp b/include/ck_tile/ops/flatmm/pipeline/flatmm_pipeline_agmem_bgmem_creg_v1_policy.hpp
old mode 100755
new mode 100644
index 31c080c520..57e58b24f0
--- a/include/ck_tile/ops/flatmm/pipeline/flatmm_pipeline_agmem_bgmem_creg_v1_policy.hpp
+++ b/include/ck_tile/ops/flatmm/pipeline/flatmm_pipeline_agmem_bgmem_creg_v1_policy.hpp
@@ -229,13 +229,13 @@ struct UniversalFlatmmPipelineAgBgCrPolicy
     template <typename Problem>
     CK_TILE_HOST_DEVICE static constexpr index_t GetSmemPackA()
     {
-        using A         = remove_cvref_t<typename Problem::ADataType>;
-        using BlockGemm = remove_cvref_t<decltype(GetBlockGemm<Problem>())>;
+        using A           = remove_cvref_t<typename Problem::ADataType>;
+        using BlockFlatmm = remove_cvref_t<decltype(GetBlockFlatmm<Problem>())>;
 
-        constexpr index_t KPack    = static_cast<index_t>(BlockGemm::Traits::KPack);
-        constexpr index_t VecElems = static_cast<index_t>(Problem::VectorLoadSize / sizeof(A));
+        constexpr index_t KPack    = BlockFlatmm::BlockPolicy::WarpGemm::kKPerThread;
+        constexpr index_t VecElems = Problem::VectorLoadSize / sizeof(A);
 
-        return (KPack < VecElems) ? KPack : VecElems;
+        return min(KPack, VecElems);
     }
 
     template <typename Problem>