From a94d67d44b6147789be5e2453cfa068614a6507b Mon Sep 17 00:00:00 2001 From: Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com> Date: Mon, 23 Mar 2026 12:08:00 +0800 Subject: [PATCH] =?UTF-8?q?[SKILL]=20fix(bench):=20Support=20model-specifi?= =?UTF-8?q?c=20DenoisingStage=20variants=20in=E2=80=A6=20(#21137)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../diffusion-kernel/scripts/bench_diffusion_denoise.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/python/sglang/multimodal_gen/.claude/skills/diffusion-kernel/scripts/bench_diffusion_denoise.py b/python/sglang/multimodal_gen/.claude/skills/diffusion-kernel/scripts/bench_diffusion_denoise.py index 2d67c15e2..852fe5fcf 100755 --- a/python/sglang/multimodal_gen/.claude/skills/diffusion-kernel/scripts/bench_diffusion_denoise.py +++ b/python/sglang/multimodal_gen/.claude/skills/diffusion-kernel/scripts/bench_diffusion_denoise.py @@ -323,12 +323,16 @@ def run_benchmark_once( 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}, ...] denoise_latency_s = None for step in perf.get("steps", []): + step_name = step.get("name") if ( - step.get("name") == "DenoisingStage" + isinstance(step_name, str) + and "DenoisingStage" in step_name and step.get("duration_ms") is not None ): denoise_latency_s = float(step["duration_ms"]) / 1000.0