pytorch_pfn_extras.profiler.TimeSummary.report¶
- TimeSummary.report(tag, use_cuda=False)¶
Context manager to automatically report execution times.
The start and completion times are obtained automatically, the user only needs to provide a tag to identify the value in the summary values.
- Parameters
tag (str) – A name to identify the section of code being profiled.
use_cuda (bool) – Indicates if GPU time should also be profiled.
- Return type
Generator[pytorch_pfn_extras.profiler._time_summary._ReportNotification, None, None]