From 567d5925fe896a7b70043271f5621d6fe6e650d4 Mon Sep 17 00:00:00 2001 From: Cheng Wan <54331508+ch-wan@users.noreply.github.com> Date: Sat, 19 Sep 2026 00:43:56 -0700 Subject: [PATCH] Fix mxfp4 padding test stubbing an accessor the module no longer imports (#40308) --- .../unit/layers/quantization/test_mxfp4_trtllm_padding.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test/registered/unit/layers/quantization/test_mxfp4_trtllm_padding.py b/test/registered/unit/layers/quantization/test_mxfp4_trtllm_padding.py index 1497cbadd..72fd7855d 100644 --- a/test/registered/unit/layers/quantization/test_mxfp4_trtllm_padding.py +++ b/test/registered/unit/layers/quantization/test_mxfp4_trtllm_padding.py @@ -11,6 +11,7 @@ import torch from sglang.srt.layers.moe.token_dispatcher import StandardDispatchOutput from sglang.srt.layers.moe.topk import StandardTopKOutput 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.test_utils import CustomTestCase @@ -88,7 +89,7 @@ class TestMxfp4TrtllmPadding(CustomTestCase): shards.append(prepare(shard)) 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, "use_symmetric_memory", return_value=nullcontext()), ):