torch.cuda.amp
- 若API未标明支持情况则代表该API暂无验证结果，待验证后更新表格。
- 在使用支持的cuda接口时，需要将API名称中的cuda变换为NPU形式才能使用：torch.cuda.***变换为torch_npu.npu.***。举例如下：
torch.cuda.amp.autocast --> torch_npu.npu.amp.autocast
API名称 |
NPU形式名称 |
是否支持 |
限制与说明 |
|---|---|---|---|
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.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 |
是 |