自动迁移操作简单,且修改内容少,只需在训练脚本中导入库代码即可完成脚本迁移。
import torch import torch_npu ..... from torch_npu.contrib import transfer_to_npu
此操作仅涉及PyTorch 1.11.0版本,PyTorch 2.1.0及以上版本已经接入原生to_cpu逻辑,无需执行此操作。
cd pytorch/torch_npu/csrc/aten vi npu_native_functions.yaml
在“tocpu”配置下添加算子API名称。
tocpu: - angle - mode - nanmedian.dim_values - nansum - native_dropout - native_dropout_backward - poisson - vdot - view_as_complex - view_as_real