[diffusion] fix: fix stages not logged when perf_dump_path is provided (#16016)

python/sglang/multimodal_gen/runtime/managers/gpu_worker.py

@@ -9,6 +9,7 @@ from typing import List
 import torch
 from setproctitle import setproctitle
 
+from sglang.multimodal_gen import envs
 from sglang.multimodal_gen.runtime.distributed import (
     get_sp_group,
     maybe_init_distributed_environment_and_model_parallel,
@@ -31,10 +32,7 @@ from sglang.multimodal_gen.runtime.utils.logging_utils import (
     globally_suppress_loggers,
     init_logger,
 )
-from sglang.multimodal_gen.runtime.utils.perf_logger import (
-    PerformanceLogger,
-    StageProfiler,
-)
+from sglang.multimodal_gen.runtime.utils.perf_logger import PerformanceLogger
 
 logger = init_logger(__name__)
 
@@ -129,7 +127,8 @@ class GPUWorker:
             duration_ms = (time.monotonic() - start_time) * 1000
             output_batch.timings.total_duration_ms = duration_ms
 
-            if StageProfiler.metrics_enabled():
+            # TODO: extract to avoid duplication
+            if req.perf_dump_path is not None or envs.SGLANG_DIFFUSION_STAGE_LOGGING:
                 PerformanceLogger.log_request_summary(timings=output_batch.timings)
         except Exception as e:
             logger.error(