From 9daab9664de0af4361f0edb367b0a20c7bd8cacc Mon Sep 17 00:00:00 2001 From: "assistant-librarian[bot]" Date: Wed, 10 Dec 2025 16:15:45 +0000 Subject: [PATCH] Merge commit 'b15df372553e0f80a660124f1b558d9cb276bd08' into develop --- 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(