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

Fix mxfp4 padding test stubbing an accessor the module no longer imports (#40308)

Browse Source
This commit is contained in:
Cheng Wan
2026-09-19 00:43:56 -07:00
committed by GitHub
parent 3a5f52e144
commit 567d5925fe
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/unit/layers/quantization/test_mxfp4_trtllm_padding.py
+2 -1
View File
@@ -11,6 +11,7 @@ import torch
from sglang.srt.layers.moe.token_dispatcher import StandardDispatchOutput from sglang.srt.layers.moe.token_dispatcher import StandardDispatchOutput
from sglang.srt.layers.moe.topk import StandardTopKOutput from sglang.srt.layers.moe.topk import StandardTopKOutput
from sglang.srt.layers.quantization import mxfp4_flashinfer_trtllm_moe as mxfp4 from sglang.srt.layers.quantization import mxfp4_flashinfer_trtllm_moe as mxfp4
from sglang.srt.runtime_context import get_parallel
from sglang.test.ci.ci_register import register_cuda_ci from sglang.test.ci.ci_register import register_cuda_ci
from sglang.test.test_utils import CustomTestCase from sglang.test.test_utils import CustomTestCase
@@ -88,7 +89,7 @@ class TestMxfp4TrtllmPadding(CustomTestCase):
shards.append(prepare(shard)) shards.append(prepare(shard))
with ( with (
patch.object(mxfp4, "get_tp_group", return_value=None), get_parallel().override(tp_group=None),
patch.object(mxfp4, "is_allocation_symmetric", return_value=False), patch.object(mxfp4, "is_allocation_symmetric", return_value=False),
patch.object(mxfp4, "use_symmetric_memory", return_value=nullcontext()), patch.object(mxfp4, "use_symmetric_memory", return_value=nullcontext()),
): ):