minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 13 Packages Projects Releases Wiki Activity

[CI] Fix stale GPU capability test patches (#37148)

Browse Source
This commit is contained in:
YAMY
2026-08-30 10:12:32 -07:00
committed by GitHub
parent 6a9366f036
commit 9a03bc2dc3
2 changed files with 25 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/quant/test_fp8_utils.py
+21 -12
View File
@@ -73,9 +73,9 @@ class TestApplyFp8LinearScaleDispatch(CustomTestCase):
)
)
for capability in (
"_is_sm90_supported",
"_is_sm100_supported",
"_is_sm120_supported",
"is_sm90",
"is_sm100",
"is_sm120",
):
with self.subTest(capability=capability):
input, qinput, weight, input_scale, weight_scale = self._make_inputs()
@@ -92,14 +92,20 @@ class TestApplyFp8LinearScaleDispatch(CustomTestCase):
)
capabilities = {
"_is_sm90_supported": False,
"_is_sm100_supported": False,
"_is_sm120_supported": False,
"is_sm90": False,
"is_sm100": False,
"is_sm120": False,
}
capabilities[capability] = True
with patch.multiple(fp8_utils, **capabilities), patch.object(
with patch.object(
fp8_utils,
"get_platform",
return_value=SimpleNamespace(**capabilities),
), patch.object(
fp8_utils, "fp8_scaled_mm", side_effect=fake_fp8_scaled_mm
), patch.object(fp8_utils, "get_exec", return_value=exec_config):
), patch.object(
fp8_utils, "get_exec", return_value=exec_config
):
fp8_utils.apply_fp8_linear(
input,
weight,
@@ -151,11 +157,14 @@ class TestApplyFp8LinearScaleDispatch(CustomTestCase):
(mat_a.shape[0], mat_b.shape[1]), dtype=out_dtype, device=mat_a.device
)
with patch.multiple(
with patch.object(
fp8_utils,
_is_sm90_supported=False,
_is_sm100_supported=False,
_is_sm120_supported=False,
"get_platform",
return_value=SimpleNamespace(
is_sm90=False,
is_sm100=False,
is_sm120=False,
),
), patch.object(fp8_utils, "fp8_scaled_mm", side_effect=fake_fp8_scaled_mm):
fp8_utils.apply_fp8_linear(
input,