ops-transformer 算子仓当前支持的编译芯片列表(ASCEND_ALL_COMPUTE_UNIT / SOC_VERSION_LIST / SUPPORT_COMPUTE_UNIT_SHORT)不包含 Ascend5162A,无法通过 --soc=ascend5162a 编译输出 5162A 算子包。
ASCEND_ALL_COMPUTE_UNIT
SOC_VERSION_LIST
SUPPORT_COMPUTE_UNIT_SHORT
--soc=ascend5162a
使 ops-transformer 算子仓支持 Ascend5162A(DAV_5162)芯片的算子编译与交付,扩展算子仓支持的硬件范围。
在编译框架与 UT 桩代码中注册 ascend5162a 计算单元(arch 目录复用 arch35),共 8 处:
CMakeLists.txt
ARCH_DIRECTORY_LIST
ascend5162a
build.sh
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
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
CANN 自定义算子编译:build.sh --soc=ascend5162a
build.sh --soc=ascend5162a
不涉及算子逻辑改动,仅编译框架层适配。
不涉及
💡 备注:已在本地环境(CANN 9.2.0)验证 --soc=ascend5162a 编译打包通过。
/assign @wuxuening
一、背景信息
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增加ascend5162acmake/scripts/util/const_var.py与scripts/util/const_var.py:SOC_MAP_EXT增加ascend5162a → Ascend5162Acmake/scripts/util/opdesc_parser.py:SOC_TO_SHORT_SOC_MAP增加ascend5162atests/ut/framework_normal/common/op_api_csv_case_loader.h:SOC_VERSION映射增加Ascend5162Atests/ut/framework_normal/op_api/stub/opdev/platform.cpp/.h:SocVersion枚举增加ASCEND5162A及 ToString 映射3.1 使能方式
CANN 自定义算子编译:
build.sh --soc=ascend5162a3.2 总体设计
不涉及算子逻辑改动,仅编译框架层适配。
3.3 支持硬件
Ascend5162A
3.4 算子约束限制
不涉及
💡 备注:已在本地环境(CANN 9.2.0)验证
--soc=ascend5162a编译打包通过。