[diffusion] feat: support warmup with resolutions (#16330)

2026-01-05 10:16:26 +08:00
parent 0fee6bc632
commit 9a8ba3c189
12 changed files with 142 additions and 67 deletions
--- a/python/sglang/srt/model_executor/cuda_graph_runner.py
+++ b/python/sglang/srt/model_executor/cuda_graph_runner.py
@@ -534,7 +534,7 @@ class CudaGraphRunner:
        )
        graph_fn = (
            partial(memory_saver_adapter.cuda_graph, tag=GPU_MEMORY_TYPE_CUDA_GRAPH)
-            if memory_saver_adapter.enabled
+            if memory_saver_adapter.log_timing
            else self.device_module.graph
        )
        with graph_fn(cuda_graph=graph, pool=pool, stream=stream):