已关闭
[Requirement|需求建议]: NMSWithMask/SortedNMS算子新增ascend350平台支持 #896
wenqi_wu创建于  13 天前关闭于  12 天前
wenqi_wu
13 天前 创建

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 配置:

  1. 算子 CMakeLists.txt 中 SUPPORT_COMPUTE_UNIT 增加 "ascend350",SUPPORT_TILING_DIR 对应增加 "arch35"
  2. op_host *_def.cpp 中通过 this->AICore().AddConfig("ascend350", config) 复用 ascend950 的 AICore 配置
  3. 新增 op_host/config/ascend350/ 下的 *_binary.json 与 *_simplified_key.ini
  4. scripts/kernel/binary_config/ascendc_config.json 中两算子 compute_units 增加 "ascend350"
likedislike
Wwenqi_wu
13 天前 添加了label:requirement
Wwenqi_wu
13 天前 关联了pull request:NMSWithMask/SortedNMS算子新增ascend350平台支持
Zzhuzemao成员
13 天前 将 wenqi_wu 设为负责人
CANN-robotCANN-robot成员
12 天前 关闭了 issue
CANN-robotCANN-robot成员
12 天前 添加了label:resolved