[AMD] Fix CI base-a fwd_occupancy: disable SGLANG_SANITIZE_NAN_LOGITS in AMD CI (#28075)

This commit is contained in:
Michael
2026-06-12 14:14:34 -07:00
committed by GitHub
parent 1e71c1a859
commit 65d76bd3f6
2 changed files with 8 additions and 2 deletions
+4 -2
View File
@@ -5,7 +5,7 @@ hellaswag accuracy."""
import unittest
from sglang.srt.utils import kill_process_tree
from sglang.srt.utils import is_hip, kill_process_tree
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
from sglang.test.kits.basic_api_contract_kit import BasicAPIContractMixin
from sglang.test.kits.basic_decode_correctness_kit import BasicDecodeCorrectnessMixin
@@ -36,7 +36,9 @@ class TestBasicSanity(
# 5090 + Llama-3.1-8B single-batch decode with overlap scheduler +
# cuda graph measured ~99 median in CI; async-assert probes are off in
# base-a, so the threshold can sit right under the measured median.
fwd_occupancy_threshold = 99.0
# AMD also measures ~99 but with less headroom; keep ~1pp of margin
# there so small per-step changes don't flake the gate.
fwd_occupancy_threshold = 98.0 if is_hip() else 99.0
@classmethod
def setUpClass(cls):