Tiny re-introduce profile id logging (#6912)
This commit is contained in:
@@ -2221,7 +2221,7 @@ class Scheduler(
|
|||||||
) -> ProfileReqOutput | None:
|
) -> ProfileReqOutput | None:
|
||||||
stage_str = f" for {stage.__str__()}" if stage else ""
|
stage_str = f" for {stage.__str__()}" if stage else ""
|
||||||
logger.info(
|
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
|
activities = self.profiler_activities
|
||||||
|
|||||||
Reference in New Issue
Block a user