在模型运行时遇到报错“RuntimeError: could not create a primitive descriptor for a matmul primitive”

问题描述

• 报错截图

  

• 报错文本

   File "/root/alibaba/xcode-ml/pytorch/torch/nn/modules/linear.py", line 114, in forward
      return F.linear(input, self.weight, self.bias)
  RuntimeError: could not create a primitive descriptor for a matmul primitive

问题分析

PyTorch2.1.0版本执行F.linear计算，当“out feature”为“1”时，CPU计算在aarch64架构出现上述报错。由于PyTorch2.1.0自身原因导致此报错，PyTorch社区的issue单为https://github.com/pytorch/pytorch/issues/110149。

处理方法

将PyTorch2.1.0升级为PyTorch2.2.0及以上版本即可解决此问题。