为 ThresholdV2、ConcatOffset 两个算子扩展芯片支持,新增适配 ascend350、ascend960pr、ascend960dt。当前这两个算子的算子定义、编译单元与 tiling 配置均未覆盖上述芯片,导致在对应产品上无法编译部署。
ThresholdV2
ConcatOffset
ascend350
ascend960pr
ascend960dt
CANN 算子新芯片适配需求(与 PR https://gitcode.com/cann/ops-nn/pull/11280 同源)。
使 ThresholdV2、ConcatOffset 可在 ascend350、ascend960pr、ascend960dt 上运行,扩大算子的芯片/产品覆盖范围。
CMakeLists.txt
SUPPORT_COMPUTE_UNIT
arch35
_def.cpp
AICore().AddConfig
op_kernel/CMakeLists.txt
COMPUTE_UNITS
op_host/config/<chip>/*_binary.json
关联 PR:https://gitcode.com/cann/ops-nn/pull/11280
/assign @esok11
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中增加目标芯片,并复用arch35tiling 目录;_def.cpp中为目标芯片注册AICore().AddConfig;op_kernel/CMakeLists.txt中补充对应COMPUTE_UNITS(ConcatOffset 为新增该文件);op_host/config/<chip>/*_binary.jsontiling 配置。关联 PR:https://gitcode.com/cann/ops-nn/pull/11280