torch.amp
Note
若API“是否支持”为“是”，“限制与说明”为“-”，说明此API和原生API支持度保持一致。
| API名称 | NPU形式名称 | 是否支持 | 限制与说明 |
|---|---|---|---|
| torch.autocast | torch.autocast | 是 | - |
| torch.cuda.amp.autocast | torch_npu.npu.amp.autocast | 是 | - |
| torch.cuda.amp.custom_fwd | torch_npu.npu.amp.custom_fwd | 是 | - |
| torch.cuda.amp.custom_bwd | torch_npu.npu.amp.custom_bwd | 是 | - |
| torch.cpu.amp.autocast | torch.cpu.amp.autocast | 是 | - |
| torch.cuda.amp.GradScaler | torch_npu.npu.amp.GradScaler | 是 | - |