From 7a5869c5780391bf24c85abf5737352e2192c459 Mon Sep 17 00:00:00 2001 From: lirui927 Date: Mon, 8 Jun 2026 07:55:32 +0000 Subject: [PATCH] fix deepgemm_fp8_paged_mqa_logits kernel input parameter mismatch issue --- aiter/ops/triton/attention/pa_mqa_logits.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/aiter/ops/triton/attention/pa_mqa_logits.py b/aiter/ops/triton/attention/pa_mqa_logits.py index 3c785d4644..bee3fb7e6c 100644 --- a/aiter/ops/triton/attention/pa_mqa_logits.py +++ b/aiter/ops/triton/attention/pa_mqa_logits.py @@ -506,6 +506,7 @@ def deepgemm_fp8_paged_mqa_logits( ) if triton_version >= Version("3.5.0"): cdna_version = get_cdna_version() + is_gfx1250 = get_gfx() == "gfx1250" kernel[grid]( batch_size, next_n, @@ -533,6 +534,7 @@ def deepgemm_fp8_paged_mqa_logits( KVBlockSize, hidden_dim, cdna_version, + is_gfx1250, ) else: # load AOT compiled gluon kernel assert triton_version < Version(