pytorch_pfn_extras.profiler.TimeSummary.report¶
- TimeSummary.report(tag: str, use_cuda: bool = False) None ¶
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.