[feature] enable NPU CI (#7935)

Co-authored-by: Even Zhou <14368888+iforgetmyname@users.noreply.github.com>
This commit is contained in:
ronnie_zheng
2025-07-20 23:12:42 +03:00
committed by GitHub
parent 1fc455e8b6
commit 93d124ef5a
4 changed files with 113 additions and 15 deletions

View File

@@ -20,22 +20,10 @@ from sglang.test.test_utils import (
run_bench_offline_throughput,
)
DEFAULT_MODEL_NAME_FOR_TEST = "Qwen/Qwen2.5-7B-Instruct"
class TestAscendAttnBackend(CustomTestCase):
def test_latency(self):
output_throughput = run_bench_offline_throughput(
DEFAULT_MODEL_NAME_FOR_TEST,
[
"--attention-backend",
"ascend",
],
)
print(f"{output_throughput=}")
if is_in_ci():
self.assertGreater(output_throughput, 18)
def test_gsm8k(self):
model = DEFAULT_MODEL_NAME_FOR_TEST
base_url = DEFAULT_URL_FOR_TEST