From fb5c7e0314e8b4ef8a44c7fc7f62e113bae48574 Mon Sep 17 00:00:00 2001 From: Po Yen Chen Date: Wed, 10 Dec 2025 23:08:41 +0800 Subject: [PATCH] fix: python 3.8 compatibility in fmha codegen (#3388) [ROCm/composable_kernel commit: b15df372553e0f80a660124f1b558d9cb276bd08] --- example/ck_tile/01_fmha/codegen/ops/fmha_fwd.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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 c00bdcea3b..edc0e049c5 100644 --- a/example/ck_tile/01_fmha/codegen/ops/fmha_fwd.py +++ b/example/ck_tile/01_fmha/codegen/ops/fmha_fwd.py @@ -770,7 +770,7 @@ def create_kernel( class CompatibilityRuleFactory: @staticmethod - def get_rules() -> list[CompatibilityRule]: + def get_rules() -> List[CompatibilityRule]: # in group mode, spad/skpad must be true, since we can't predict if seqlen of current batch need pad or not def check_mode(problem_ctx: ProblemContext, kernel_ctx: KernelContext) -> bool: if problem_ctx.mode == "group": @@ -812,7 +812,7 @@ class CompatibilityRuleFactoryGfx9(CompatibilityRuleFactory): _AVAILABLE_PIPELINES = frozenset({"qr", "qr_async", "qs"}) @classmethod - def get_rules(cls) -> list[CompatibilityRule]: + def get_rules(cls) -> List[CompatibilityRule]: rules = CompatibilityRuleFactory.get_rules() def check_hdim_tile( @@ -846,7 +846,7 @@ class CompatibilityRuleFactoryGfx950(CompatibilityRuleFactoryGfx9): ) @classmethod - def get_rules(cls) -> list[CompatibilityRule]: + def get_rules(cls) -> List[CompatibilityRule]: rules = CompatibilityRuleFactoryGfx9.get_rules() def check_tile_pipeline(