Tiny re-introduce profile id logging (#6912)

This commit is contained in:
fzyzcjy
2025-06-07 17:32:50 +08:00
committed by GitHub
parent 9736cd3b7d
commit d5c097a2f9

View File

@@ -2221,7 +2221,7 @@ class Scheduler(
) -> ProfileReqOutput | None:
stage_str = f" for {stage.__str__()}" if stage else ""
logger.info(
f"Profiling starts{stage_str}. Traces will be saved to: {self.torch_profiler_output_dir}",
f"Profiling starts{stage_str}. Traces will be saved to: {self.torch_profiler_output_dir} (with profile id: {self.profile_id})",
)
activities = self.profiler_activities