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

[SKILL] fix(bench): Support model-specific DenoisingStage variants in… (#21137)

Browse Source
This commit is contained in:
Xiaoyu Zhang
2026-03-23 12:08:00 +08:00
committed by GitHub
parent 2b47bd3a34
commit a94d67d44b
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/multimodal_gen/.claude/skills/diffusion-kernel/scripts/bench_diffusion_denoise.py
+6 -2
View File
@@ -323,12 +323,16 @@ def run_benchmark_once(
float(total_ms) / 1000.0 if total_ms is not None else None float(total_ms) / 1000.0 if total_ms is not None else None
) )
# denoise latency: look in "steps" list for the "DenoisingStage" entry # denoise latency: accept the canonical "DenoisingStage" plus
# model-specific variants such as "MOVADenoisingStage" and
# "HeliosChunkedDenoisingStage".
# steps = [{"name": "DenoisingStage", "duration_ms": 1234.5}, ...] # steps = [{"name": "DenoisingStage", "duration_ms": 1234.5}, ...]
denoise_latency_s = None denoise_latency_s = None
for step in perf.get("steps", []): for step in perf.get("steps", []):
step_name = step.get("name")
if ( if (
step.get("name") == "DenoisingStage" isinstance(step_name, str)
and "DenoisingStage" in step_name
and step.get("duration_ms") is not None and step.get("duration_ms") is not None
): ):
denoise_latency_s = float(step["duration_ms"]) / 1000.0 denoise_latency_s = float(step["duration_ms"]) / 1000.0