在模型运行时遇到报错“RuntimeError: could not create a primitive descriptor for a matmul primitive”
2025/03/18
31
问题信息
| 问题来源 | 产品大类 | 产品子类 | 关键字 |
|---|---|---|---|
| 官方 | 模型训练 | PyTorch | -- |
问题现象描述
- 报错截图
- 报错文本
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及以上版本即可解决此问题。