PyTorch框架在ARM CPU上算子计算结果异常

问题描述

当前使用的PyTorch官方原生框架在ARM CPU上运行时，算子计算结果会出现异常，此问题为原生框架社区的已知问题，详细内容可参考PyTorch官方社区ISSUE。

解决方案

可通过以下方式解决：

• 修改算子输入数据类型，使用float64数据类型进行运算。
• 升级编译arm版本PyTorch使用的gcc编译器至9.4版本及以上，并使用相同编译器重新编译torch_npu、apex、mmcv等其他配套软件（避免因编译器版本不匹配导致兼容性问题）。