Closed
[Bug-Report|缺陷反馈]: math仓重复注册ScatterND ONNX插件导致nn仓插件失效 #4547
zhang-wenbo-beatAdd on 5 days agoClosed at 4 days ago
5 days ago Added label: bug-report
5 days ago Associated pull request
5 days ago Modify description of issue
4 days ago Assign zhang-wenbo-beat as responsible person
4 days ago Close issue
4 days ago Added label: resolved
一、问题描述 (必填)
math 仓在
common/src/framework/similar_opp_onnx_plugin.cpp中重复注册了 ONNXScatterND插件,共 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 条注册。二、环境信息 (可选)
三、重现步骤 (可选)
ScatterND节点的 ONNX 模型执行模型转换/图构建;ScatterND插件未生效(被 math 仓的注册覆盖),ScatterND节点映射异常。四、预期结果 (可选)
math 仓不再注册
ScatterND,由 nn 仓唯一注册;两仓插件共存时 nn 仓ScatterND插件正常生效,ScatterND节点可正确映射。💡 备注(选填)
修复 PR:https://gitcode.com/cann/ops-math/merge_requests/7160