Star610
419
代码
代码
Issues60
Pull Requests91
Actions
项目讨论
Wiki
分析
Star610
419
已合并
fix(example): rename NPU_ARCH to NPU_SOC_VERSION and dynamically adapt Python version tag #1067
liu-wei创建于 7月1日
fix(example): rename NPU_ARCH to NPU_SOC_VERSION and dynamically adapt Python version tag #1067
已合并
liu-wei创建于 7月1日
从liu-wei/ops-cvmaster_fast_kernel合入到cann/ops-cvmaster
liu-wei
liu-wei成员
7月1日

描述

examples/fast_kernel_launch_example/ 里有两处需要改造，不属于同一 concern 但都局限在该 example 目录，合并到同一个修复提交：

改动 A：NPU_ARCH 重命名为 NPU_SOC_VERSION

PyTorch 官方 torch_npu 的同名环境变量 NPU_ARCH 在新版本里被复用为昇腾 SOC 版本号，原本"扩展模块编译目标架构"的语义已发生迁移。继续使用旧名容易在用户 shell / CI 环境里跟 torch_npu 的同名变量直接撞车。统一改成 NPU_SOC_VERSION 更贴合"芯片型号"的语义。

为了不破坏现有用户的 CI 脚本：

  • 旧变量 NPU_ARCH 仍可读取，仅打印 deprecation 日志
  • CMake / Python 任一端能识别 NPU_SOC_VERSION 时优先使用，识别不到再回退到 NPU_ARCH

涉及文件：

  • CMakeLists.txt — NPU_ARCH 的 cache 读取逻辑替换为 NPU_SOC_VERSION + fallback
  • cmake/func.cmake — recursive_add_subdirectory 引用改 NPU_SOC_VERSION
  • setup.py — CMakeBuildCommand 的环境变量读取迁移，并加 deprecation 日志
  • README.md — 文档同步

改动 B：wheel tag 从硬编码 cp38 改为跟随实际 Python 版本

ABI3Wheel.get_tag() 之前硬编码 python = "cp38"，导致用 Python 3.9+ 构建时 wheel 文件名仍标 cp38（PEP 425 平台标签机制会因"环境 Python 版本 vs wheel 声称版本"不一致，pip 安装时直接报 platform-tag-implicit-metadata 类错误）。

改为 f"cp3{sys.version_info.minor}" 跟随 sys.version_info.minor，让产物跟实际构建环境一致。

涉及文件：setup.py（ABI3Wheel.get_tag()）

关联的Issue

无

测试

在容器内（cp39 / cp310 / cp311）分别执行：

cd examples/fast_kernel_launch_example
python3 -m build --wheel -n
# 验证产物文件名 cp 标签与 python3 --version 一致
ls dist/*.whl
python3 -c "import sys; print(sys.version_info.minor)"  # 对照确认

回退验证：保留旧 NPU_ARCH 环境变量时，构建应继续成功，并在 setup.py 日志里看到 "NPU_ARCH is deprecated, please use NPU_SOC_VERSION."。

文档更新

README.md 已同步：新增 NPU_SOC_VERSION 三种芯片档位说明（Atlas A2 → ascend910b、Atlas A3 → ascend910_93、Ascend 950PR/950DT → ascend950），并标注旧 NPU_ARCH 已 deprecated 但仍兼容。

类型标签

likedislike
Pull Request已成功合入, 合并人@CANN-robot
（感谢 liu-wei 的贡献）
liu-weiliu-wei成员
7月1日 创建了 pull request，commit 17722c44
CANN-robotCANN-robot成员
7月1日 添加了label：cann-cla/yes
CANN-robot
CANN-robot成员
7月1日 评论：

Thanks for your pull-request.
The full list of commands accepted by me can be found at here。
You can get sig-info at here

PR Approval Progress

✅ Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

module lgtm status approve status
*/*/README.md ✅ 张磊, 杜慧萍 (2/2) ✅ 杜慧萍 (1/1)
repo-cann/ops-cv ✅ 张磊, 王林木 (2/2) ✅ 王林木, 张磊 (2/1)

💡 Tip:

  • Committer can comment /approve or /lgtm
  • Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

liu-wei, thanks for your pull request. All authors of the commits have signed the CLA. 👍

likedislike
atomgit-bot
atomgit-bot
7月1日 评论：

变更摘要

此 PR 主要对 examples/fast_kernel_launch_example 下的构建配置进行两项改进：将环境变量 NPU_ARCH 重命名为 NPU_SOC_VERSION（并保留向后兼容），以及将 Python wheel 的版本标签从硬编码的 cp38 改为根据当前解释器动态生成。

主要改动

  • 环境变量重命名 NPU_ARCH → NPU_SOC_VERSION：在 cmake/func.cmake 的 recursive_add_subdirectory 宏中将目录路径变量从 ${NPU_ARCH} 改为 ${NPU_SOC_VERSION}；在 setup.py 的 CMakeBuildCommand 中将 CMake 参数从 -DNPU_ARCH= 改为 -DNPU_SOC_VERSION=，并优先读取 NPU_SOC_VERSION 环境变量。
  • NPU_ARCH 向后兼容支持：当 NPU_SOC_VERSION 未设置时，setup.py 会回退读取旧的 NPU_ARCH 环境变量，若使用了旧变量名则输出 deprecation 日志提示。
  • Python wheel 版本标签动态适配：在 ABI3Wheel.get_tag() 中将 Python 标签从硬编码的 "cp38" 改为 f"cp3{sys.version_info.minor}"，使构建的 wheel 标签自动匹配当前 Python 解释器的次版本号，并为此新增 import sys。
likedislike
不准确？
atomgit-bot
atomgit-bot
7月1日 评论：

代码审查

审查总结

本 Diff 共涉及 4 个文件的变更，审查结论如下：

文件 审查结果
examples/fast_kernel_launch_example/CMakeLists.txt 无问题
examples/fast_kernel_launch_example/README.md 无问题
examples/fast_kernel_launch_example/cmake/func.cmake 无问题
examples/fast_kernel_launch_example/setup.py P1 问题 1 个

发现统计：

  • P0：0
  • P1：1
  • P2：0
  • P3：0

整体风险判定：中高。 setup.py 中 ABI3Wheel.get_tag() 的 python tag 动态化改动会导致在 Python ≥ 3.9 环境下构建的 abi3 轮子无法被 Python 3.8–3.9 的 pip 安装，与项目声明的 Python ≥ 3.8 支持契约相悖。建议在合入前修复该项。

类型 数量
🔴 阻塞 1
🟡 建议 0

⛔ 需要修改

likedislike
不准确？
atomgit-bot
atomgit-bot7月1日进行代码检视3
examples/fast_kernel_launch_example/setup.py
已过期
@@ -69,3 +70,3 @@
6970 def get_tag(self):
7071 python, abi, plat = super().get_tag()
71- python = "cp38"
72+ python = f"cp3{sys.version_info.minor}"
atomgit-bot
atomgit-bot7月1日评论：

🟠 High Priority

变更点：setup.py 第 72 行，ABI3Wheel.get_tag() 方法中，原先硬编码的 python = "cp38" 被改为 python = f"cp3{sys.version_info.minor}"。

影响的行为/契约：对于 abi3 标签的 wheel，根据 PEP 425 及 pip 的兼容性判定规则，python tag 表示最低支持的 Python 版本，而非构建环境的 Python 版本。cp38-abi3 表示 "CPython 3.8 及更高版本"，而 cp311-abi3 表示 "CPython 3.11 及更高版本"。

失败模式：该项目在 CMakeLists.txt 第 94 行设置了 Py_LIMITED_API=0x03080000（即 Python 3.8 稳定 ABI），二进制实际上兼容 Python ≥ 3.8。但若在 Python 3.11 环境构建，wheel tag 会变为 cp311-abi3，pip 在 Python 3.8–3.10 上将拒绝安装此 wheel，因为 tag 声明的 python 最低版本为 3.11。这与类文档字符串 "this wheel supports multiple python versions >= 3.8" 以及 python_requires='>=3.8' 的元数据相矛盾。

触发条件：使用 Python ≥ 3.9 的环境执行 python3 -m build --wheel -n 构建。

建议：将 python tag 恢复为固定常量 "cp38"（与 Py_LIMITED_API=0x03080000 保持一致），或根据 Py_LIMITED_API 定义动态推导出正确的最低版本号，而不是使用构建时 Python 的 minor 版本。

likedislike
不准确？
liu-wei
liu-wei成员
7月1日 评论：

example不存在发包，不存在A版本 python环境编出来，到B版本python环境用

System
系统消息系统
7月1日 评论：

changed this line on f729fadf view diff detail

liu-wei
liu-wei成员
7月1日 评论：

compile

likedislike
liu-weiliu-wei成员
7月1日 预合并成功（commit_id: 2a20999f968cf387ba4e8506ded3aa24aedb3738）
CANN-robotCANN-robot成员
7月1日 添加了label：ci-pipeline-running
CANN-robot
CANN-robot成员
7月1日 评论：

流水线任务触发成功
任务链接 [2ccd89145f6d49e2b0921ed4b2c3da25][流水线指导]

任务名称状态日志下载链接
Compile_Ascend_X86 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_mobile_station ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_experimental ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_experimental ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_A5 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_A5 ✅ SUCCESS >>>>> >>>>>
pre_comment ✅ SUCCESS >>>>>
Compile_Pre ✅ SUCCESS >>>>>
Compile_Ascend_X86_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_mobile_station_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_experimental_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_A5_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_experimental_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_A5_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_910b ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_910c ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_950 ✅ SUCCESS >>>>> >>>>>
Compile_classify ✅ SUCCESS >>>>>
UT_Test ✅ SUCCESS
PreSmoke_A900 ✅ SUCCESS >>>>>
API_Check ✅ SUCCESS >>>>>
PreSmoke_ATK_Test_A2 ✅ SUCCESS >>>>>
UT_Test_report_lcov ✅ SUCCESS

[2026-07-01 10:49:31]    CI执行结束

likedislike
CANN-robot
CANN-robot成员
7月1日 评论：

流水线任务触发成功
任务链接 [e192e2c5d0444ff6a55b41a6c905de51][流水线指导]

任务名称状态日志下载链接
codecheck ✅ SUCCESS >>>>>
SCA ✅ SUCCESS >>>>>
antipoison ✅ SUCCESS >>>>>
codecheck_checkpr ✅ SUCCESS
StaticCheck_codespell_check ✅ SUCCESS
StaticCheck_link_validity_check ✅ SUCCESS
StaticCheck_resource_existence_check ✅ SUCCESS
StaticCheck_tag_closed_check ✅ SUCCESS
StaticCheck_markdownlint ✅ SUCCESS
codecheck_codestyle ✅ SUCCESS >>>>>
codecheck_precommit ✅ SUCCESS >>>>>

[2026-07-01 10:46:24]    CI执行结束

likedislike
CANN-robotCANN-robot成员
7月1日 添加了label：api-check-pass
此处折叠了6条事件消息 查看更多
CANN-robotCANN-robot成员
7月1日 删除了label：ci-pipeline-passed
CANN-robot
CANN-robot成员
7月1日 评论：

Notification

This pull request has been changed(code update) or closed, so removes the following label(s): ci-pipeline-passed.

likedislike
liu-wei
liu-wei成员
7月1日 评论：

compile

likedislike
liu-weiliu-wei成员
7月1日 预合并成功（commit_id: 3e86ae738dda1faf1db647748680651c550e60b6）
CANN-robotCANN-robot成员
7月1日 删除了label：api-check-pass
CANN-robotCANN-robot成员
7月1日 添加了label：ci-pipeline-running
CANN-robot
CANN-robot成员
7月1日 评论：

流水线任务触发成功
任务链接 [37771a2b8a6a434cab2cce21aa7b9dba][流水线指导]

任务名称状态日志下载链接
Compile_Ascend_X86 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_mobile_station ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_experimental ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_experimental ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_A5 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_A5 ✅ SUCCESS >>>>> >>>>>
pre_comment ✅ SUCCESS >>>>>
Compile_Pre ✅ SUCCESS >>>>>
Compile_Ascend_X86_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_mobile_station_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_experimental_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_A5_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_experimental_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_A5_ubuntu24 ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_910b ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_910c ✅ SUCCESS >>>>> >>>>>
Compile_X86_monitor_950 ✅ SUCCESS >>>>> >>>>>
Compile_classify ✅ SUCCESS >>>>>
UT_Test ✅ SUCCESS
PreSmoke_A900 ✅ SUCCESS >>>>>
API_Check ✅ SUCCESS >>>>>
PreSmoke_ATK_Test_A2 ✅ SUCCESS >>>>>
UT_Test_report_lcov ✅ SUCCESS

[2026-07-01 11:17:21]    CI执行结束

likedislike
CANN-robot
CANN-robot成员
7月1日 评论：

流水线任务触发成功
任务链接 [d968932d8cca4378b458ccafdf63793d][流水线指导]

任务名称状态日志下载链接
codecheck ✅ SUCCESS >>>>>
SCA ✅ SUCCESS >>>>>
antipoison ✅ SUCCESS >>>>>
codecheck_checkpr ✅ SUCCESS
StaticCheck_codespell_check ✅ SUCCESS
StaticCheck_link_validity_check ✅ SUCCESS
StaticCheck_resource_existence_check ✅ SUCCESS
StaticCheck_tag_closed_check ✅ SUCCESS
StaticCheck_markdownlint ✅ SUCCESS
codecheck_codestyle ✅ SUCCESS >>>>>
codecheck_precommit ✅ SUCCESS >>>>>

[2026-07-01 11:13:44]    CI执行结束

likedislike
CANN-robotCANN-robot成员
7月1日 添加了label：api-check-pass
liu-weiliu-wei成员
7月1日 修改了pull request 的描述
CANN-robotCANN-robot成员
7月1日 删除了label：ci-pipeline-running
CANN-robotCANN-robot成员
7月1日 添加了label：ci-pipeline-passed
liu-wei
liu-wei成员
7月2日 评论：

🔗 关联 follow-up issue：#589

本 PR 重命名 NPU_ARCH → NPU_SOC_VERSION 时保留了兼容读 + deprecation 日志，但没有给完全移除的时间表。#589 跟踪移除 NPU_ARCH 兼容层的 release 计划，建议在下次 minor release（合入 NPU_SOC_VERSION 唯一支持的版本）实施。

具体需要清理的位置、联动扫其他 example 的检查、每次 milestone 的验收 checklist 都在 issue 正文里。

likedislike
liu-weiliu-wei成员
7月2日 关联了issue：[Follow-up] NPU_ARCH deprecation timeline — 计划在哪个 release 完全移除
ZhaiPeiChao
ZhaiPeiChao成员
7月2日 评论：

/lgtm
/approve

likedislike
CANN-robot
CANN-robot成员
7月2日 评论：

The following users do not have permission to comment /lgtm or /approve on any module in this PR:
ZhaiPeiChao

likedislike
CANN-robot
CANN-robot成员
7月2日 评论：

The following users do not have permission to comment /lgtm or /approve on any module in this PR:
ZhaiPeiChao

likedislike
L
llimwang成员
7月2日 评论：

/approve

likedislike
CANN-robot
CANN-robot成员
7月2日 评论：

The following users do not have permission to comment /lgtm or /approve on any module in this PR:
ZhaiPeiChao

likedislike
Z
zl_hw成员
7月2日 评论：

/lgtm
/approve

likedislike
CANN-robot
CANN-robot成员
7月2日 评论：

The following users do not have permission to comment /lgtm or /approve on any module in this PR:
ZhaiPeiChao

likedislike
CANN-robot
CANN-robot成员
7月2日 评论：

The following users do not have permission to comment /lgtm or /approve on any module in this PR:
ZhaiPeiChao

likedislike
G
gitee-duhuiping成员
7月2日 评论：

/approve

likedislike
CANN-robot
CANN-robot成员
7月2日 评论：

The following users do not have permission to comment /lgtm or /approve on any module in this PR:
ZhaiPeiChao

likedislike
CANN-robotCANN-robot成员
7月2日 添加了label：lgtmapproved
CANN-robotCANN-robot成员
7月2日 关闭了关联的issue
CANN-robotCANN-robot成员
7月2日 合入了pull request