From e254cdf326849e893a8129b53d969e2aa45f88c4 Mon Sep 17 00:00:00 2001 From: Yuwei An Date: Tue, 23 Dec 2025 21:28:05 -0800 Subject: [PATCH] [CI] Remove pcg-omni-ci (#15656) --- .../test_piecewise_cuda_graph_2_gpu.py | 44 +------------------ 1 file changed, 1 insertion(+), 43 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 b99b5d9ef..40fe6656c 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 @@ -12,49 +12,7 @@ from sglang.test.test_utils import ( ) # CI Registration - 2-GPU tests (80GB GPUs required) -register_cuda_ci(est_time=255, suite="stage-b-test-large-2-gpu") - - -class TestPiecewiseCudaGraphQwen3OmniMOE(CustomTestCase): - """Test piecewise CUDA graph with Qwen3-Omni-30B-A3B-Instruct model""" - - @classmethod - def setUpClass(cls): - cls.model = "Qwen/Qwen3-Omni-30B-A3B-Instruct" - cls.base_url = DEFAULT_URL_FOR_TEST - cls.process = popen_launch_server( - cls.model, - cls.base_url, - timeout=DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH, - other_args=[ - "--enable-piecewise-cuda-graph", - "--piecewise-cuda-graph-compiler", - "eager", - "--disable-radix-cache", - "--tp=2", - ], - ) - - @classmethod - def tearDownClass(cls): - kill_process_tree(cls.process.pid) - - def test_gsm8k_accuracy(self): - """Test GSM8K accuracy with 8-shot setting""" - num_examples = 2000 - - args = SimpleNamespace( - base_url=self.base_url, - model=self.model, - eval_name="mgsm_en", - num_examples=num_examples, - num_threads=min(num_examples, 1024), - ) - - metrics = run_eval(args) - print(f"GSM8K Accuracy: {metrics['score']:.3f}") - - self.assertGreaterEqual(metrics["score"], 0.70) +register_cuda_ci(est_time=160, suite="stage-b-test-large-2-gpu") class TestPiecewiseCudaGraphFusedMoE(CustomTestCase):