序号 |
API名称 |
npu对应API名称 |
是否支持 |
|---|---|---|---|
1 |
torch.cuda.amp.autocast |
torch_npu.npu.amp.autocast |
是 |
2 |
torch.cuda.amp.custom_fwd |
torch_npu.npu.amp.custom_fwd |
否 |
3 |
torch.cuda.amp.custom_bwd |
torch_npu.npu.amp.custom_bwd |
否 |
4 |
torch.cuda.amp.GradScaler |
torch_npu.npu.amp.GradScaler |
是 |