Closed
[Bug-Report|缺陷反馈]: math仓重复注册ScatterND ONNX插件导致nn仓插件失效 #4547
zhang-wenbo-beatAdd on  5 days agoClosed at  4 days ago
zhang-wenbo-beat
zhang-wenbo-beatCollaborators
5 days ago Create

一、问题描述 (必填)

math 仓在 common/src/framework/similar_opp_onnx_plugin.cpp 中重复注册了 ONNX ScatterND 插件,共 10 条(opset 8/10/11/12/13/14/15/16/17/18),全部映射到 TensorScatterUpdate:

REGISTER_CUSTOM_ONNX_OP(ai.onnx::8::ScatterND, TensorScatterUpdate, ScatterNDV8);
REGISTER_CUSTOM_ONNX_OP(ai.onnx::10::ScatterND, TensorScatterUpdate, ScatterNDV10);
REGISTER_CUSTOM_ONNX_OP(ai.onnx::11::ScatterND, TensorScatterUpdate, ScatterNDV11);
REGISTER_CUSTOM_ONNX_OP(ai.onnx::12::ScatterND, TensorScatterUpdate, ScatterNDV12);
REGISTER_CUSTOM_ONNX_OP(ai.onnx::13::ScatterND, TensorScatterUpdate, ScatterNDV13);
REGISTER_CUSTOM_ONNX_OP(ai.onnx::14::ScatterND, TensorScatterUpdate, ScatterNDV14);
REGISTER_CUSTOM_ONNX_OP(ai.onnx::15::ScatterND, TensorScatterUpdate, ScatterNDV15);
REGISTER_CUSTOM_ONNX_OP(ai.onnx::16::ScatterND, TensorScatterUpdate, ScatterNDV16);
REGISTER_CUSTOM_ONNX_OP(ai.onnx::17::ScatterND, TensorScatterUpdate, ScatterNDV17);
REGISTER_CUSTOM_ONNX_OP(ai.onnx::18::ScatterND, TensorScatterUpdate, ScatterNDV18);

nn 仓同样注册了 ONNX ScatterND 插件。两仓插件同时加载时发生重复注册冲突,导致 nn 仓的 ScatterND 插件失效,ONNX 模型中的 ScatterND 节点无法按 nn 仓的实现完成映射/转换。

ScatterND 属于 nn 仓职责范围,math 仓不应重复注册,建议删除 math 仓上述全部 10 条注册。

二、环境信息 (可选)

  • 昇腾硬件型号:不涉及(插件注册冲突,与硬件无关)
  • CANN版本:master 分支(CANN 9.x)
  • 操作系统:Linux
  • 其他相关软件版本:ops-math 与 ops-nn ONNX 插件共存

三、重现步骤 (可选)

  1. 同时部署并加载 ops-math 与 ops-nn 的 ONNX 插件;
  2. 使用包含 ScatterND 节点的 ONNX 模型执行模型转换/图构建;
  3. 观察 nn 仓 ScatterND 插件未生效(被 math 仓的注册覆盖),ScatterND 节点映射异常。

四、预期结果 (可选)

math 仓不再注册 ScatterND,由 nn 仓唯一注册;两仓插件共存时 nn 仓 ScatterND 插件正常生效,ScatterND 节点可正确映射。

💡 备注(选填)

修复 PR:https://gitcode.com/cann/ops-math/merge_requests/7160

likedislike
zhang-wenbo-beatzhang-wenbo-beatCollaborators
5 days ago Added label: bug-report
zhang-wenbo-beatzhang-wenbo-beatCollaborators
5 days ago Associated pull request
zhang-wenbo-beatzhang-wenbo-beatCollaborators
5 days ago Modify description of issue
陈思陈思Collaborators
4 days ago Assign zhang-wenbo-beat as responsible person
CANN-robotCANN-robotCollaborators
4 days ago Close issue
CANN-robotCANN-robotCollaborators
4 days ago Added label: resolved