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}'