From 3c24d690a1d1787545a46cb7759fb6961547e9cc Mon Sep 17 00:00:00 2001 From: aska-0096 Date: Thu, 29 May 2025 10:13:50 +0000 Subject: [PATCH] remove single rate mfma restriction for f8 blockscale gemm --- .../grid/gridwise_gemm_xdl_cshuffle_v3_multi_d_ab_scale.hpp | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_v3_multi_d_ab_scale.hpp b/include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_v3_multi_d_ab_scale.hpp index ef84dd182a..e03e60db25 100644 --- a/include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_v3_multi_d_ab_scale.hpp +++ b/include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_v3_multi_d_ab_scale.hpp @@ -153,9 +153,7 @@ struct GridwiseGemmMultiD_ABScale_xdl_cshuffle_v3 static constexpr bool is_single_rate_mfma = (((is_same::value || is_same::value) && lcm_AK1_BK1 <= 4) || - (is_same::value && lcm_AK1_BK1 <= 8) || - ((is_same::value || is_same::value) && - lcm_AK1_BK1 < 32)) + (is_same::value && lcm_AK1_BK1 <= 8)) ? true : false; static constexpr auto is_scale_mfma = false;