From 609fdf3322708f57af8c06ca66fcfe227d8233da Mon Sep 17 00:00:00 2001 From: Tianyuan Wu Date: Mon, 18 Aug 2025 16:45:40 +0800 Subject: [PATCH] Fix CI build error (#2695) Signed-off-by: Tianyuan Wu [ROCm/composable_kernel commit: 7310830d143836ea10eee42121d2e4e43647d1e1] --- .../ck_tile/ops/gemm/warp/warp_gemm_attribute_wmma_impl.hpp | 3 +++ .../gemm/warp/warp_gemm_attribute_wmma_impl_base_traits.hpp | 6 ++++++ 2 files changed, 9 insertions(+) diff --git a/include/ck_tile/ops/gemm/warp/warp_gemm_attribute_wmma_impl.hpp b/include/ck_tile/ops/gemm/warp/warp_gemm_attribute_wmma_impl.hpp index 13727d41b1..88fde40067 100644 --- a/include/ck_tile/ops/gemm/warp/warp_gemm_attribute_wmma_impl.hpp +++ b/include/ck_tile/ops/gemm/warp/warp_gemm_attribute_wmma_impl.hpp @@ -35,6 +35,9 @@ struct WarpGemmAttributeWmmaImpl static constexpr index_t kN = Traits::kN; static constexpr index_t kK = Traits::kK; + static constexpr index_t kAMBlock = Traits::kAMBlock; + static constexpr index_t kBNBlock = Traits::kBNBlock; + static constexpr index_t kRepeat = Traits::kRepeat; static constexpr index_t kAMLane = Traits::kAMLane; static constexpr index_t kBNLane = Traits::kBNLane; diff --git a/include/ck_tile/ops/gemm/warp/warp_gemm_attribute_wmma_impl_base_traits.hpp b/include/ck_tile/ops/gemm/warp/warp_gemm_attribute_wmma_impl_base_traits.hpp index 7ea5507d09..7a3190e6f4 100644 --- a/include/ck_tile/ops/gemm/warp/warp_gemm_attribute_wmma_impl_base_traits.hpp +++ b/include/ck_tile/ops/gemm/warp/warp_gemm_attribute_wmma_impl_base_traits.hpp @@ -22,6 +22,9 @@ struct WmmaTraitsBase static constexpr index_t kN = 16; static constexpr index_t kK = 16; + static constexpr index_t kAMBlock = 1; + static constexpr index_t kBNBlock = 1; + static constexpr index_t kRepeat = 2; static constexpr index_t kAMLane = 16; static constexpr index_t kBNLane = 16; @@ -61,6 +64,9 @@ struct WmmaTraitsBase static constexpr index_t kN = 16; static constexpr index_t kK = 16; + static constexpr index_t kAMBlock = 1; + static constexpr index_t kBNBlock = 1; + static constexpr index_t kRepeat = 1; static constexpr index_t kAMLane = 16; static constexpr index_t kBNLane = 16;