From c8d6e24eb513ff4d60bc0ab883ede9a2c75ea298 Mon Sep 17 00:00:00 2001 From: fsx950223 Date: Fri, 16 May 2025 09:41:27 +0000 Subject: [PATCH] change condition --- include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp b/include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp index 04a2a57c58..316c9d26f7 100644 --- a/include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp +++ b/include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp @@ -1055,7 +1055,7 @@ struct FmhaFwdKernel // # of required blocks is different in each groups, terminate unnecessary blocks // earlier - if(kargs.seqlen_q <= i_m0 && kargs.seqlen_q == 1) + if(kargs.seqlen_q <= i_m0 || kargs.seqlen_q == 1) { return; }