Classes 4
emit_itt
classFull reference ↗- class tensorplay.profiler.profiler.emit_itt(enabled=True, record_shapes=False)[source]
Emit instrumentation ranges for active operations.
emit_nvtx
classFull reference ↗- class tensorplay.profiler.profiler.emit_nvtx(enabled=True, record_shapes=False)[source]
Emit named ranges for active operations.
profile
classFull reference ↗- class tensorplay.profiler.profiler.profile(enabled=True, *, activities=None, schedule=None, on_trace_ready=None, record_shapes=False, profile_memory=False, with_stack=False, with_flops=False, with_modules=False, use_device=None, gpu_timing=False, gpu_trace=False, with_samples=False)[source]
Context manager that records dispatched operations and annotations.
- step()[source]
Advance the configured step schedule.
record_function
classFull reference ↗- class tensorplay.profiler.profiler.record_function(name, args=None)[source]
Mark a user-defined span in the active profiling session.
