若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 |
是 |
|
torch.cuda.amp.GradScaler.get_backoff_factor |
torch_npu.npu.amp.GradScaler.get_backoff_factor |
是 |
|
torch.cuda.amp.GradScaler.get_growth_factor |
torch_npu.npu.amp.GradScaler.get_growth_factor |
是 |
|
torch.cuda.amp.GradScaler.get_growth_interval |
torch_npu.npu.amp.GradScaler.get_growth_interval |
是 |
|
torch.cuda.amp.GradScaler.get_scale |
torch_npu.npu.amp.GradScaler.get_scale |
是 |
|
torch.cuda.amp.GradScaler.is_enabled |
torch_npu.npu.amp.GradScaler.is_enabled |
是 |
|
torch.cuda.amp.GradScaler.load_state_dict |
torch_npu.npu.amp.GradScaler.load_state_dict |
是 |
|
torch.cuda.amp.GradScaler.scale |
torch_npu.npu.amp.GradScaler.scale |
是 |
|
torch.cuda.amp.GradScaler.set_backoff_factor |
torch_npu.npu.amp.GradScaler.set_backoff_factor |
是 |
|
torch.cuda.amp.GradScaler.set_growth_factor |
torch_npu.npu.amp.GradScaler.set_growth_factor |
是 |
|
torch.cuda.amp.GradScaler.set_growth_interval |
torch_npu.npu.amp.GradScaler.set_growth_interval |
是 |
|
torch.cuda.amp.GradScaler.state_dict |
torch_npu.npu.amp.GradScaler.state_dict |
是 |
|
torch.cuda.amp.GradScaler.step |
torch_npu.npu.amp.GradScaler.step |
是 |
|
torch.cuda.amp.GradScaler.unscale_ |
torch_npu.npu.amp.GradScaler.unscale_ |
是 |
|
torch.cuda.amp.GradScaler.update |
torch_npu.npu.amp.GradScaler.update |
是 |