Skip to content

Commit b2c70ef

Browse files
author
root
committed
test(musa): add platform-specific benchmark smoke
1 parent 7792a7c commit b2c70ef

2 files changed

Lines changed: 31 additions & 1 deletion

File tree

tests/benchmarks/configs/musa.yaml

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
serve:
2+
- name: serve_smoke_dummy
3+
model: qwen3
4+
case: 4b_tp2_eager_musa
5+
server_parameters:
6+
served_model_name: smoke-model
7+
tensor_parallel_size: 1
8+
max_model_len: 1024
9+
gpu_memory_utilization: 0.7
10+
max_num_seqs: 4
11+
enforce_eager: true
12+
disable_custom_all_reduce: true
13+
load_format: dummy
14+
client_parameters:
15+
model: smoke-model
16+
backend: openai-chat
17+
endpoint: /v1/chat/completions
18+
dataset_name: random
19+
random_input_len: 32
20+
random_output_len: 4
21+
num_prompts: 5
22+
percentile_metrics: ttft,tpot,itl,e2el
23+
goodput: e2el:10000

tests/run.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -350,7 +350,14 @@ def _discover_benchmark_tests(self) -> list[TestCase]:
350350
else:
351351
selected_types = set(selected_smoke)
352352

353-
config_path = Path("tests/benchmarks/configs/smoke.yaml")
353+
platform_config_path = Path(
354+
f"tests/benchmarks/configs/{self.config.platform}.yaml"
355+
)
356+
config_path = (
357+
platform_config_path
358+
if platform_config_path.exists()
359+
else Path("tests/benchmarks/configs/smoke.yaml")
360+
)
354361
if not config_path.exists():
355362
print(f"[run] Warning: benchmark config not found: {config_path}")
356363
return []

0 commit comments

Comments
 (0)