[diffusion] CI: fix CI test case skip problem (#15874)

Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
Prozac614
2025-12-26 19:42:20 +08:00
committed by GitHub
parent 5d421db883
commit 3778c2fc6d
5 changed files with 44 additions and 10 deletions

View File

@@ -26,7 +26,10 @@ from sglang.multimodal_gen.runtime.utils.logging_utils import (
globally_suppress_loggers,
init_logger,
)
from sglang.multimodal_gen.runtime.utils.perf_logger import PerformanceLogger
from sglang.multimodal_gen.runtime.utils.perf_logger import (
PerformanceLogger,
StageProfiler,
)
logger = init_logger(__name__)
@@ -108,7 +111,7 @@ class GPUWorker:
if output_batch.timings:
duration_ms = (time.monotonic() - start_time) * 1000
output_batch.timings.total_duration_ms = duration_ms
if req.perf_dump_path is not None:
if StageProfiler.metrics_enabled():
PerformanceLogger.log_request_summary(timings=output_batch.timings)
except Exception as e:
logger.error(