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 |
是 |