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

[CI][RFC] Replace black-jupyter with ruff-format (#37210)

Browse Source 
Co-authored-by: Alison Shao <a.shao@wustl.edu>
This commit is contained in:
Alex Nails
2026-09-02 19:46:08 -07:00
committed by GitHub
co-authored by Alison Shao
parent 2641e427be
commit 28262c20df
1411 changed files with 7766 additions and 8176 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/quant/test_fp8_utils.py
+27 -22
View File
@@ -37,12 +37,12 @@ class TestInverseTransformScaleUe8m0(CustomTestCase):
sf_packed_recreated = transform_scale_ue8m0(sf_fp32_recreated, mn=mn)
assert torch.all(
sf_packed_original == sf_packed_recreated
), f"{sf_packed_original=} {sf_packed_recreated}"
assert torch.all(
sf_fp32_original == sf_fp32_recreated
), f"{sf_fp32_original=} {sf_fp32_recreated}"
assert torch.all(sf_packed_original == sf_packed_recreated), (
f"{sf_packed_original=} {sf_packed_recreated}"
)
assert torch.all(sf_fp32_original == sf_fp32_recreated), (
f"{sf_fp32_original=} {sf_fp32_recreated}"
)
class TestApplyFp8LinearScaleDispatch(CustomTestCase):
@@ -97,14 +97,16 @@ class TestApplyFp8LinearScaleDispatch(CustomTestCase):
"is_sm120": False,
}
capabilities[capability] = True
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
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),
):
fp8_utils.apply_fp8_linear(
input,
@@ -157,15 +159,18 @@ class TestApplyFp8LinearScaleDispatch(CustomTestCase):
(mat_a.shape[0], mat_b.shape[1]), dtype=out_dtype, device=mat_a.device
)
with patch.object(
fp8_utils,
"get_platform",
return_value=SimpleNamespace(
is_sm90=False,
is_sm100=False,
is_sm120=False,
with (
patch.object(
fp8_utils,
"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):
patch.object(fp8_utils, "fp8_scaled_mm", side_effect=fake_fp8_scaled_mm),
):
fp8_utils.apply_fp8_linear(
input,
weight,