From 2a8d5493f3a4c53f5e9ba457234e87d072f2b829 Mon Sep 17 00:00:00 2001 From: Yuwei An Date: Tue, 6 Jan 2026 18:57:15 -0800 Subject: [PATCH] PCG Unit Test Adjustment (#16609) Signed-off-by: Oasis-Git --- .../cuda_graph/test_piecewise_cuda_graph_2_gpu.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/test/registered/cuda_graph/test_piecewise_cuda_graph_2_gpu.py b/test/registered/cuda_graph/test_piecewise_cuda_graph_2_gpu.py index 40fe6656c..ac8176210 100644 --- a/test/registered/cuda_graph/test_piecewise_cuda_graph_2_gpu.py +++ b/test/registered/cuda_graph/test_piecewise_cuda_graph_2_gpu.py @@ -15,8 +15,8 @@ from sglang.test.test_utils import ( register_cuda_ci(est_time=160, suite="stage-b-test-large-2-gpu") -class TestPiecewiseCudaGraphFusedMoE(CustomTestCase): - """Test piecewise CUDA graph with FusedMoE Backend""" +class TestPiecewiseCudaGraphTP(CustomTestCase): + """Test piecewise CUDA graph with normal TP""" @classmethod def setUpClass(cls): @@ -32,8 +32,6 @@ class TestPiecewiseCudaGraphFusedMoE(CustomTestCase): "eager", "--tp", "2", - "--ep-size", - "2", ], )