onnxruntime-cann 模型/算子支持需求 - slanet-plus
收藏回复举报
onnxruntime-cann 模型/算子支持需求 - slanet-plus
t('forum.solved') 已解决
新人帖
发表于2025-10-09 08:21:54
0 查看

## 问题描述

当前CANN ExecutionProvider不支持ONNX的Loop算子,导致无法运行包含Loop算子的模型,如广泛使用的SLANet-plus表格识别模型。

## 具体场景

- **模型**: SLANet-plus.onnx (百度PaddleOCR开源的表格结构识别模型)

- **模型来源**: https://github.com/RapidAI/RapidTable ; https://www.modelscope.cn/models/RapidAI/RapidTable/files

- **错误信息**: `[ONNXRuntimeError] : 1 : FAIL : Node (Loop.0) Op (Loop) [ShapeInferenceError] Loop op is not supported`

## 影响范围

1. **表格识别场景**: 文档AI、OCR应用中的表格结构识别

2. **用户群体**: 使用昇腾NPU进行文档处理的开发者

3. **模型生态**: 影响PaddleOCR生态在昇腾平台的应用

## 期望结果

希望CANN能够支持ONNX Loop算子 以及 add 广播(bs, 256)+(256),使SLANet-plus等包含Loop的模型能在昇腾NPU上正常运行。

## 相关信息

- CANN版本: 8.1.RC1

- ONNX Runtime版本: onnxruntime-cann     1.22.2

- 模型ONNX版本: ai.onnx v14

- Loop算子规格: 包含32个输入,30个输出,235个内部节点

我要发帖子