[CI] Collapse the EAGLE launch matrix and the scoring engine boots on the per-commit runners (#33756)
This commit is contained in:
@@ -3,9 +3,8 @@ import os
|
||||
import re
|
||||
import unittest
|
||||
|
||||
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
|
||||
from sglang.test.ci.ci_register import register_amd_ci
|
||||
|
||||
register_cuda_ci(est_time=103, stage="base-b", runner_config="1-gpu-small")
|
||||
register_amd_ci(est_time=106, suite="stage-b-test-1-gpu-small-amd-mi35x")
|
||||
import time
|
||||
from types import SimpleNamespace
|
||||
@@ -14,7 +13,7 @@ import requests
|
||||
import torch
|
||||
|
||||
from sglang.srt.utils import kill_process_tree
|
||||
from sglang.srt.utils.common import is_cuda_alike, mxfp_supported
|
||||
from sglang.srt.utils.common import is_cuda_alike, is_gfx95_supported
|
||||
from sglang.test.few_shot_gsm8k import run_eval
|
||||
from sglang.test.test_utils import (
|
||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||
@@ -36,7 +35,7 @@ class TestOnlineQuantizationMemoryLoad(CustomTestCase):
|
||||
f"test requires {cls.tp} devices, only {torch.cuda.device_count()} are available."
|
||||
)
|
||||
|
||||
if not mxfp_supported():
|
||||
if not is_gfx95_supported():
|
||||
raise unittest.SkipTest(
|
||||
"online MXFP4 quantization requires an AMD ROCm device with "
|
||||
"FP4 hardware support (gfx95x, e.g. MI355x)"
|
||||
|
||||
Reference in New Issue
Block a user