[NPU] [CI] temporarily disable mtp test (#17614)

test/srt/ascend/test_ascend_deepseek_mtp.py

@@ -1,94 +0,0 @@
-import unittest
-from types import SimpleNamespace
-from urllib.parse import urlparse
-
-from sglang.srt.environ import envs
-from sglang.srt.utils import kill_process_tree
-from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
-from sglang.test.test_utils import (
-    DEFAULT_URL_FOR_TEST,
-    CustomTestCase,
-    popen_launch_server,
-)
-
-TEST_MODEL_MATRIX = {
-    "/root/.cache/modelscope/hub/models/vllm-ascend/DeepSeek-R1-0528-W8A8": {
-        "accuracy": 0.95,
-        "latency": 1000,
-        "output_throughput": 6,
-    },
-}
-
-
-class TestAscendDeepSeekMTP(CustomTestCase):
-
-    @classmethod
-    def setUpClass(cls):
-        cls.models = TEST_MODEL_MATRIX.keys()
-        cls.base_url = DEFAULT_URL_FOR_TEST
-        cls.url = urlparse(DEFAULT_URL_FOR_TEST)
-
-        cls.common_args = [
-            "--trust-remote-code",
-            "--attention-backend",
-            "ascend",
-            "--mem-fraction-static",
-            0.8,
-            "--disable-radix-cache",
-            "--chunked-prefill-size",
-            32768,
-            "--tp-size",
-            16,
-            "--dp-size",
-            2,
-            "--enable-dp-attention",
-            "--speculative-algorithm",
-            "NEXTN",
-            "--speculative-num-steps",
-            1,
-            "--speculative-eagle-topk",
-            1,
-            "--speculative-num-draft-tokens",
-            2,
-        ]
-
-        envs.SGLANG_NPU_USE_MLAPO.set(True)
-        envs.SGLANG_ENABLE_SPEC_V2.set(True)
-        envs.SGLANG_ENABLE_OVERLAP_PLAN_STREAM.set(True)
-
-    def test_a_gsm8k(self):
-        for model in self.models:
-            with self.subTest(model=model):
-                print(f"##=== Testing accuracy: {model} ===##")
-
-                process = popen_launch_server(
-                    model,
-                    self.base_url,
-                    timeout=2400,
-                    other_args=[
-                        *self.common_args,
-                    ],
-                )
-
-                try:
-                    args = SimpleNamespace(
-                        num_shots=5,
-                        data_path=None,
-                        num_questions=1319,
-                        max_new_tokens=512,
-                        parallel=128,
-                        host=f"http://{self.url.hostname}",
-                        port=int(self.url.port),
-                    )
-
-                    metrics = run_eval_few_shot_gsm8k(args)
-                    self.assertGreaterEqual(
-                        metrics["accuracy"],
-                        TEST_MODEL_MATRIX[model]["accuracy"],
-                    )
-                finally:
-                    kill_process_tree(process.pid)
-
-
-if __name__ == "__main__":
-    unittest.main()

test/srt/ascend/test_ascend_tp2_fia_bf16.py

@@ -9,9 +9,7 @@ from sglang.test.test_utils import (
     DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
     DEFAULT_URL_FOR_TEST,
     CustomTestCase,
-    is_in_ci,
     popen_launch_server,
-    run_bench_offline_throughput,
 )
 
 TEST_MODEL_MATRIX = {
@@ -76,26 +74,6 @@ class TestAscendTp2Bf16(CustomTestCase):
                 finally:
                     kill_process_tree(process.pid)
 
-    def test_b_throughput(self):
-        for model in self.models:
-            with self.subTest(model=model):
-                print(f"##=== Testing throughput: {model} ===##")
-
-                output_throughput = run_bench_offline_throughput(
-                    model,
-                    [
-                        *self.common_args,
-                    ],
-                )
-
-                print(f"##=== {model} throughput: {output_throughput} ===##")
-
-                if is_in_ci():
-                    self.assertGreater(
-                        output_throughput,
-                        TEST_MODEL_MATRIX[model]["output_throughput"],
-                    )
-
 
 if __name__ == "__main__":
     unittest.main()

test/srt/run_suite.py

@@ -153,7 +153,7 @@ suite_ascend = {
     ],
     "per-commit-16-npu-a3": [
         TestFile("ascend/test_ascend_deepep.py", 3600),
-        TestFile("ascend/test_ascend_deepseek_mtp.py", 2800),
+        # TestFile("ascend/test_ascend_deepseek_mtp.py", 2800),
         TestFile("ascend/test_ascend_w4a4_quantization.py", 600),
     ],
 }