Closed
polar算子适配ascend960dt/ascend960pr #4434
yu_qinfeiAdd on  11 days agoClosed at  11 days ago
yu_qinfeiCollaborators
11 days ago Create

问题描述

Polar 算子当前仅在 ascend950 平台注册(SUPPORT_COMPUTE_UNIT/AICore().AddConfig/二进制配置均只有 ascend950),Ascend 960 系列(ascend960dt/ascend960pr)上缺失该算子能力。

期望结果

Polar 算子在 ascend960dt/ascend960pr 平台可用:

  1. 平台注册:math/polar/op_host/polar_def.cpp 增加 ascend960dt/ascend960pr 的 AICore().AddConfig;
  2. 编译配置:math/polar/CMakeLists.txt 的 SUPPORT_COMPUTE_UNIT 追加两平台,tiling 目录沿用 arch35;
  3. 二进制配置:补齐 op_host/config/ascend960dt/、op_host/config/ascend960pr/ 下的 polar_binary.json 与 polar_simplified_key.ini(内容沿用 ascend950)。

约束

  • 输入 abs/angle:FLOAT,ND;输出 y:COMPLEX64,ND,支持广播;
  • 无 kernel 代码改动,复用现有 arch35 kernel 与 tiling。

关联PR

cann/ops-math!7051

likedislike
CANN-robotCANN-robotCollaborators
11 days ago Close issue
CANN-robotCANN-robotCollaborators
11 days ago Added label: resolved