[AMD] CI: fix Lean decode crash on the EAGLE path (#37119)

Co-authored-by: Bingxu Chen <bingxche@amd.com>
2026-09-03 23:04:21 -07:00
parent 7825e5ffca
commit 4e756ecc4a
1 changed files with 1 additions and 0 deletions
@@ -1203,6 +1203,7 @@ def decode_attention_fwd(
        and _should_use_lean_decode(
            enable_lean, logit_cap, sinks, xai_temperature_len, score_mod
        )
        and lean_locks is not None
    ):
        total_programs, XCD_REMAP, NUM_XCDS = _lean_decode_launch_params(
            v_buffer.shape[-2], kv_group_num