From 1d3304ab9ed41d3ef5e656a95b6d506bd796dc9d Mon Sep 17 00:00:00 2001 From: Tianxing Wu Date: Fri, 7 Nov 2025 12:31:31 +0000 Subject: [PATCH] revert change on fmha --- example/ck_tile/01_fmha/fmha_fwd_v3_impl.hpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/example/ck_tile/01_fmha/fmha_fwd_v3_impl.hpp b/example/ck_tile/01_fmha/fmha_fwd_v3_impl.hpp index 457139fbff..194675f962 100644 --- a/example/ck_tile/01_fmha/fmha_fwd_v3_impl.hpp +++ b/example/ck_tile/01_fmha/fmha_fwd_v3_impl.hpp @@ -57,8 +57,9 @@ struct fmha_fwd_v3_kernel_traits { static constexpr auto date_type = DataType; static constexpr bool is_variable_seqlen = IsVariableSeqlen; - static constexpr bool is_masking = IsMasking - // M0 N0 K0 N1 K1 + static constexpr bool is_masking = IsMasking; + + // M0 N0 K0 N1 K1 using fmha_block_tile = sequence<256, 32, 128, 128, 32, 128>; using fmha_warp_gemm_shape = sequence<32, 32, 16>; using fmha_block_warps = sequence<8, 1, 1>;