amp
torch_npu.npu.get_amp_supported_dtype
torch_npu.npu.is_autocast_enabled
torch_npu.npu.set_autocast_enabled
torch_npu.npu.get_autocast_dtype
torch_npu.npu.set_autocast_dtype
父主题：
torch_npu.npu