chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[diffusion] CI: add script for automatically generation ci perf baseline (#16389)

Browse Source
This commit is contained in:
Xiaoyu Zhang
2026-01-05 13:18:35 +08:00
committed by GitHub
parent 561a3e04f6
commit 520c048d55
4 changed files with 330 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
python/sglang/multimodal_gen/docs/ci_perf.md Normal file
View File

@@ -0,0 +1,30 @@
## Perf baseline generation script
`python/sglang/multimodal_gen/test/scripts/gen_perf_baselines.py` starts a local diffusion server, issues requests for selected test cases, aggregates stage/denoise-step/E2E timings from the perf log, and writes the results back to the `scenarios` section of `perf_baselines.json`.
### Usage
Update a single case:
```bash
python python/sglang/multimodal_gen/test/scripts/gen_perf_baselines.py --case qwen_image_t2i
```
Select by regex:
```bash
python python/sglang/multimodal_gen/test/scripts/gen_perf_baselines.py --match 'qwen_image_.*'
```
Run all keys from the baseline file `scenarios`:
```bash
python python/sglang/multimodal_gen/test/scripts/gen_perf_baselines.py --all-from-baseline
```
Specify input/output paths and timeout:
```bash
python python/sglang/multimodal_gen/test/scripts/gen_perf_baselines.py --baseline python/sglang/multimodal_gen/test/server/perf_baselines.json --out /tmp/perf_baselines.json --timeout 600
```