PCG Unit Test Adjustment (#16609)
Signed-off-by: Oasis-Git <ayw.sirius19@gmail.com>
This commit is contained in:
@@ -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",
|
||||
],
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user