import torch_npu from torch_npu.contrib import transfer_to_npu
import torch_mlu sed -i 's/cuda/mlu/g' *.py