torch_npu.npu.amp

Torch_npu提供amp相关的部分接口，具体可参考torch.cuda.amp。