ops-cv 算子仓当前支持的编译芯片列表(ASCEND_ALL_COMPUTE_UNIT / SUPPORT_COMPUTE_UNIT_SHORT)不包含 Ascend5162A,无法通过 --soc=ascend5162a 编译输出 5162A 算子包。
ASCEND_ALL_COMPUTE_UNIT
SUPPORT_COMPUTE_UNIT_SHORT
--soc=ascend5162a
使 ops-cv 算子仓支持 Ascend5162A(DAV_5162)芯片的算子编译与交付,扩展算子仓支持的硬件范围。
在编译框架与 UT 桩代码中注册 ascend5162a 计算单元,共 7 处:
CMakeLists.txt
ascend5162a
scripts/build_conf.sh
scripts/kernel/binary_script/build_env.sh
SOC_MAP
Ascend5162A
scripts/util/const_var.py
SOC_MAP_EXT
ascend5162a → Ascend5162A
scripts/util/opdesc_parser.py
SOC_TO_SHORT_SOC_MAP
tests/ut/op_api/stub/opdev/platform.cpp/.h
SocVersion
ASCEND5162A
CANN 自定义算子编译:build.sh --soc=ascend5162a
build.sh --soc=ascend5162a
不涉及算子逻辑改动,仅编译框架层适配。
不涉及
💡 备注:已在本地环境(CANN 9.2.0)验证 --soc=ascend5162a 编译打包通过。
一、背景信息
ops-cv 算子仓当前支持的编译芯片列表(
ASCEND_ALL_COMPUTE_UNIT/SUPPORT_COMPUTE_UNIT_SHORT)不包含 Ascend5162A,无法通过--soc=ascend5162a编译输出 5162A 算子包。二、价值/作用
使 ops-cv 算子仓支持 Ascend5162A(DAV_5162)芯片的算子编译与交付,扩展算子仓支持的硬件范围。
三、设计方案
在编译框架与 UT 桩代码中注册 ascend5162a 计算单元,共 7 处:
CMakeLists.txt:ASCEND_ALL_COMPUTE_UNIT增加ascend5162ascripts/build_conf.sh:SUPPORT_COMPUTE_UNIT_SHORT增加ascend5162ascripts/kernel/binary_script/build_env.sh:SOC_MAP增加Ascend5162A映射scripts/util/const_var.py:SOC_MAP_EXT增加ascend5162a → Ascend5162Ascripts/util/opdesc_parser.py:SOC_TO_SHORT_SOC_MAP增加ascend5162atests/ut/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编译打包通过。