[NPU] [CI] temporarily disable mtp test (#17614)
This commit is contained in:
@@ -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()
|
||||
|
||||
@@ -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),
|
||||
],
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user