From 534b95c05c93b0adc89652b525c5b3210384f24b Mon Sep 17 00:00:00 2001 From: carlushuang Date: Fri, 20 Dec 2024 16:40:45 +0800 Subject: [PATCH] hot-fix (#1768) [ROCm/composable_kernel commit: 1c45ca35dd5c215e0c1db1f40f01556f467f52a8] --- .../ck_tile/ops/gemm/warp/warp_gemm_attribute_mfma_impl.hpp | 3 +++ 1 file changed, 3 insertions(+) diff --git a/include/ck_tile/ops/gemm/warp/warp_gemm_attribute_mfma_impl.hpp b/include/ck_tile/ops/gemm/warp/warp_gemm_attribute_mfma_impl.hpp index fa24711de2..21a865e792 100644 --- a/include/ck_tile/ops/gemm/warp/warp_gemm_attribute_mfma_impl.hpp +++ b/include/ck_tile/ops/gemm/warp/warp_gemm_attribute_mfma_impl.hpp @@ -839,6 +839,9 @@ struct WarpGemmAttributeMfmaImpl_i32_32x32x16_i8 static constexpr index_t kN = 32; static constexpr index_t kK = 16; + static constexpr index_t kAMBlock = 1; + static constexpr index_t kBNBlock = 1; + static constexpr index_t kAMLane = 32; static constexpr index_t kBNLane = 32; static constexpr index_t kABKLane = 2;