API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.autograd.profiler.profile |
否 |
|
torch.autograd.profiler.emit_nvtx |
否 |
|
torch.autograd.backward |
是 |
支持bf16，fp16，fp32，fp64 不支持稀疏张量 |
profile.export_chrome_trace |
否 |
|
profile.key_averages |
否 |
|
profile.total_average |
否 |
|
torch.autograd.profiler.load_nvprof |
否 |