torch.amp
若API“是否支持”为“是”，“限制与说明”为空，说明此API和原生API支持度保持一致。
API名称 |
NPU形式名称 |
是否支持 |
限制与说明 |
|---|---|---|---|
torch.autocast |
- |
是 |
|
torch.cuda.amp.autocast |
torch_npu.npu.amp.autocast |
是 |
|
torch.cuda.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 |
是 |
父主题： PyTorch原生API支持度