torch.profiler
Note
若API“是否支持”为“是”,“限制与说明”为“-”,说明此API和原生API支持度保持一致。
| API名称 | 是否支持 | 限制与说明 |
|---|---|---|
| torch.profiler._KinetoProfile | 是 | - |
| torch.profiler._KinetoProfile.add_metadata | 是 | - |
| torch.profiler._KinetoProfile.add_metadata_json | 是 | - |
| torch.profiler._KinetoProfile.events | 否 | - |
| torch.profiler._KinetoProfile.export_chrome_trace | 是 | - |
| torch.profiler._KinetoProfile.export_memory_timeline | 是 | - |
| torch.profiler._KinetoProfile.export_stacks | 是 | - |
| torch.profiler._KinetoProfile.key_averages | 否 | - |
| torch.profiler.profile | 是 | - |
| torch.profiler.profile.step | 是 | - |
| torch.profiler.ProfilerAction | 是 | - |
| torch.profiler.ProfilerActivity | 是 | - |
| torch.profiler.ProfilerActivity.name | 是 | - |
| torch.profiler.schedule | 是 | - |
| torch.profiler.tensorboard_trace_handler | 是 | - |
| torch.profiler.itt.is_available | 否 | - |
| torch.profiler.itt.mark | 否 | - |
| torch.profiler.itt.range_push | 否 | - |
| torch.profiler.itt.range_pop | 否 | - |