若API未标明支持情况则代表该API暂无验证结果，待验证后更新表格。
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.autograd.Function |
是 |
|
torch.autograd.profiler.profile |
是 |
采集NPU上的Profiling数据时，“use_device”需设置为“npu” |
torch.autograd.profiler.emit_nvtx |
||
torch.autograd.profiler.emit_itt |
||
torch.autograd.detect_anomaly |
是 |
|
torch.autograd.set_detect_anomaly |
是 |
|
torch.autograd.graph.saved_tensors_hooks |
是 |
|
torch.autograd.graph.save_on_cpu |
是 |
|
torch.autograd.graph.disable_saved_tensors_hooks |
是 |
|
torch.autograd.graph.register_multi_grad_hook |
是 |
|
torch.autograd.graph.allow_mutation_on_saved_tensors |
是 |
支持fp32 |
torch.autograd.backward |
是 |
支持bf16，fp16，fp32，fp64 不支持稀疏张量 |
torch.autograd.grad |
是 |
|
torch.autograd.forward_ad.dual_level |
是 |
|
torch.autograd.forward_ad.make_dual |
是 |
支持fp32 |
torch.autograd.forward_ad.unpack_dual |
是 |
支持fp32 |
torch.autograd.functional.jacobian |
是 |
支持fp32 |
torch.autograd.functional.hessian |
是 |
支持fp32 |
torch.autograd.functional.vjp |
是 |
支持fp32 |
torch.autograd.functional.jvp |
是 |
支持fp32 |
torch.autograd.functional.vhp |
是 |
支持fp32 |
torch.autograd.functional.hvp |
是 |
支持fp32 |
Function.forward |
||
Function.backward |
||
Function.jvp |
||
Function.vmap |
||
FunctionCtx.mark_dirty |
是 |
|
FunctionCtx.mark_non_differentiable |
是 |
|
FunctionCtx.save_for_backward |
是 |
|
FunctionCtx.set_materialize_grads |
是 |
|
torch.autograd.gradcheck |
是 |
|
torch.autograd.gradgradcheck |
是 |
|
profile.export_chrome_trace |
是 |
|
profile.key_averages |
是 |
|
profile.self_cpu_time_total |
是 |
|
profile.total_average |
是 |
|
torch.autograd.profiler.load_nvprof |
||
Node.name |
||
Node.metadata |
||
Node.next_functions |
||
Node.register_hook |
||
Node.register_prehook |