[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_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||||
DEFAULT_URL_FOR_TEST,
|
DEFAULT_URL_FOR_TEST,
|
||||||
CustomTestCase,
|
CustomTestCase,
|
||||||
is_in_ci,
|
|
||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
run_bench_offline_throughput,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
TEST_MODEL_MATRIX = {
|
TEST_MODEL_MATRIX = {
|
||||||
@@ -76,26 +74,6 @@ class TestAscendTp2Bf16(CustomTestCase):
|
|||||||
finally:
|
finally:
|
||||||
kill_process_tree(process.pid)
|
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__":
|
if __name__ == "__main__":
|
||||||
unittest.main()
|
unittest.main()
|
||||||
|
|||||||
@@ -153,7 +153,7 @@ suite_ascend = {
|
|||||||
],
|
],
|
||||||
"per-commit-16-npu-a3": [
|
"per-commit-16-npu-a3": [
|
||||||
TestFile("ascend/test_ascend_deepep.py", 3600),
|
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),
|
TestFile("ascend/test_ascend_w4a4_quantization.py", 600),
|
||||||
],
|
],
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user