From 3a50a9b871031f91e810264afd9fd64fe24baebf Mon Sep 17 00:00:00 2001 From: Linjun-AMD Date: Wed, 15 Oct 2025 16:13:19 +0800 Subject: [PATCH] Update fmha_fwd.py --- example/ck_tile/01_fmha/codegen/ops/fmha_fwd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/example/ck_tile/01_fmha/codegen/ops/fmha_fwd.py b/example/ck_tile/01_fmha/codegen/ops/fmha_fwd.py index 6fc808c3ef..d36ce1a245 100644 --- a/example/ck_tile/01_fmha/codegen/ops/fmha_fwd.py +++ b/example/ck_tile/01_fmha/codegen/ops/fmha_fwd.py @@ -251,7 +251,7 @@ class FmhaFwdApiTrait: else: assert False @property - def seqtune(self) -> str: + def seqtune(self, max_bm0 : int) -> str: if self.bm0 in [128, 64]: return 'true/*fall back to largest tile*/' # group mode only generate spad/skpad == true else: return f'a.seqlen_q <= {self.bm0}'