pytorch_pfn_extras.profiler#

Functions

pytorch_pfn_extras.profiler.clear_tracer()

Resets the status of the global tracer.

pytorch_pfn_extras.profiler.enable_global_trace(enable)

Enable or disable tracing for all the threads.

pytorch_pfn_extras.profiler.enable_thread_trace(enable)

Enable or disable tracing for the current thread.

pytorch_pfn_extras.profiler.get_time_summary()

pytorch_pfn_extras.profiler.get_tracer([...])

Gets the current global tracer.

pytorch_pfn_extras.profiler.load_chrome_trace_as_json(...)

pytorch_pfn_extras.profiler.record(tag[, ...])

pytorch_pfn_extras.profiler.record_function(tag)

pytorch_pfn_extras.profiler.record_iterable(...)

Classes

pytorch_pfn_extras.profiler.ChromeTracer([...])

Tracer object that outputs a timeline in Chrome format.

pytorch_pfn_extras.profiler.TimeSummary(*[, ...])

Online summarization of execution times.

pytorch_pfn_extras.profiler.TraceableDataset(...)

Utility class to trace a Dataset inside the DataLoader worker threads.

pytorch_pfn_extras.profiler.Tracer()