已关闭
[Requirement|需求建议]: 算子仓适配Ascend5162A芯片编译 #5356
wuxuening创建于  17 天前关闭于  16 天前
wuxuening
17 天前 创建

一、背景信息

ops-transformer 算子仓当前支持的编译芯片列表(ASCEND_ALL_COMPUTE_UNIT / SOC_VERSION_LIST / SUPPORT_COMPUTE_UNIT_SHORT)不包含 Ascend5162A,无法通过 --soc=ascend5162a 编译输出 5162A 算子包。

二、价值/作用

使 ops-transformer 算子仓支持 Ascend5162A(DAV_5162)芯片的算子编译与交付,扩展算子仓支持的硬件范围。

三、设计方案

在编译框架与 UT 桩代码中注册 ascend5162a 计算单元(arch 目录复用 arch35),共 8 处:

  • CMakeLists.txt:ASCEND_ALL_COMPUTE_UNIT、SOC_VERSION_LIST、ARCH_DIRECTORY_LIST 增加 ascend5162a(对应 arch35)
  • build.sh:SUPPORT_COMPUTE_UNIT_SHORT 增加 ascend5162a
  • cmake/scripts/util/const_var.py 与 scripts/util/const_var.py:SOC_MAP_EXT 增加 ascend5162a → Ascend5162A
  • cmake/scripts/util/opdesc_parser.py:SOC_TO_SHORT_SOC_MAP 增加 ascend5162a
  • tests/ut/framework_normal/common/op_api_csv_case_loader.h:SOC_VERSION 映射增加 Ascend5162A
  • tests/ut/framework_normal/op_api/stub/opdev/platform.cpp/.h:SocVersion 枚举增加 ASCEND5162A 及 ToString 映射
3.1 使能方式

CANN 自定义算子编译:build.sh --soc=ascend5162a

3.2 总体设计

不涉及算子逻辑改动,仅编译框架层适配。

3.3 支持硬件

Ascend5162A

3.4 算子约束限制

不涉及

💡 备注:已在本地环境(CANN 9.2.0)验证 --soc=ascend5162a 编译打包通过。

likedislike
Wwuxuening
17 天前 添加了label:requirement
Wwuxuening
17 天前 关联了pull request:feat: 算子仓适配Ascend5162A芯片编译
游震成员
17 天前 评论:

/assign @wuxuening

likedislike
CANN-robotCANN-robot成员
17 天前 将 wuxuening 设为负责人
CANN-robotCANN-robot成员
16 天前 关闭了 issue
CANN-robotCANN-robot成员
16 天前 添加了label:resolved