已关闭
[Requirement|需求建议]: ThresholdV2、ConcatOffset 算子适配 ascend350/ascend960pr/ascend960dt #6258
esok11创建于  29 天前关闭于  19 天前
esok11成员
29 天前 创建

Backgroud(背景信息)

为 ThresholdV2、ConcatOffset 两个算子扩展芯片支持,新增适配 ascend350、ascend960pr、ascend960dt。当前这两个算子的算子定义、编译单元与 tiling 配置均未覆盖上述芯片,导致在对应产品上无法编译部署。

Origin(信息来源)

CANN 算子新芯片适配需求(与 PR https://gitcode.com/cann/ops-nn/pull/11280 同源)。

Benefit / Necessity (价值/作用)

使 ThresholdV2、ConcatOffset 可在 ascend350、ascend960pr、ascend960dt 上运行,扩大算子的芯片/产品覆盖范围。

Design(设计方案)

  • 在算子 CMakeLists.txt 的 SUPPORT_COMPUTE_UNIT 中增加目标芯片,并复用 arch35 tiling 目录;
  • 在算子 _def.cpp 中为目标芯片注册 AICore().AddConfig;
  • 在 op_kernel/CMakeLists.txt 中补充对应 COMPUTE_UNITS(ConcatOffset 为新增该文件);
  • 为各新增芯片补充 op_host/config/<chip>/*_binary.json tiling 配置。

关联 PR:https://gitcode.com/cann/ops-nn/pull/11280

likedislike
Eesok11成员
29 天前 添加了label:requirement
esok11成员
29 天前 评论:

/assign @esok11

likedislike
CANN-robotCANN-robot成员
29 天前 将 esok11 设为负责人
CANN-robotCANN-robot成员
19 天前 关闭了 issue
CANN-robotCANN-robot成员
18 天前 添加了label:resolved