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

PCG Unit Test Adjustment (#16609)

Browse Source 
Signed-off-by: Oasis-Git <ayw.sirius19@gmail.com>
This commit is contained in:
Yuwei An
2026-01-07 10:57:15 +08:00
committed by GitHub
parent 90eac38a12
commit 2a8d5493f3
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/cuda_graph/test_piecewise_cuda_graph_2_gpu.py
+2 -4
View File
@@ -15,8 +15,8 @@ from sglang.test.test_utils import (
register_cuda_ci(est_time=160, suite="stage-b-test-large-2-gpu") register_cuda_ci(est_time=160, suite="stage-b-test-large-2-gpu")
class TestPiecewiseCudaGraphFusedMoE(CustomTestCase): class TestPiecewiseCudaGraphTP(CustomTestCase):
"""Test piecewise CUDA graph with FusedMoE Backend""" """Test piecewise CUDA graph with normal TP"""
@classmethod @classmethod
def setUpClass(cls): def setUpClass(cls):
@@ -32,8 +32,6 @@ class TestPiecewiseCudaGraphFusedMoE(CustomTestCase):
"eager", "eager",
"--tp", "--tp",
"2", "2",
"--ep-size",
"2",
], ],
) )