torch_npu.npu.set_autocast_enabled(bool)
是否在设备上使能AMP。
bool：入参为True时，在设备上使能AMP，否则，不使能AMP。
torch_npu.npu.set_autocast_enabled(True)