NMSWithMask 和 SortedNMS 算子当前仅支持 ascend910b/ascend950 平台。ascend350 与 ascend950 同属 arch35(dav_3510)架构,具备复用已有算子配置的条件,但算子定义、CMakeLists 及二进制配置均未注册 ascend350,导致算子无法在 ascend350 平台上完成算子二进制与 Kernel 的匹配。
参考 ops-nn 仓已有实践:sigmoid/max_pool_v3 等算子新增 ascend350 平台支持(https://gitcode.com/cann/ops-nn/pull/9765),ops-cv 仓算子需对齐支持。
使 NMSWithMask/SortedNMS 可运行于 ascend350 平台,满足目标检测/NMS 场景在 ascend350 上的部署需求。
复用 ascend950 的 arch35 tiling 与 AICore 配置:
Backgroud(背景信息)
NMSWithMask 和 SortedNMS 算子当前仅支持 ascend910b/ascend950 平台。ascend350 与 ascend950 同属 arch35(dav_3510)架构,具备复用已有算子配置的条件,但算子定义、CMakeLists 及二进制配置均未注册 ascend350,导致算子无法在 ascend350 平台上完成算子二进制与 Kernel 的匹配。
Origin(信息来源)
参考 ops-nn 仓已有实践:sigmoid/max_pool_v3 等算子新增 ascend350 平台支持(https://gitcode.com/cann/ops-nn/pull/9765),ops-cv 仓算子需对齐支持。
Benefit / Necessity (价值/作用)
使 NMSWithMask/SortedNMS 可运行于 ascend350 平台,满足目标检测/NMS 场景在 ascend350 上的部署需求。
Design(设计方案)
复用 ascend950 的 arch35 tiling 与 AICore 配置: