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>;