已合并
[v2.9.0][bugfix]cann and pta header mixing bulid bugfix #44992
Dring创建于 16 天前
[v2.9.0][bugfix]cann and pta header mixing bulid bugfix #44992
已合并
Pull Request已成功合入, 合并人@ascend-robot
（感谢 Dring 的贡献）
ascend-robot
16 天前 评论：
16 天前 评论：
atomgit-bot
16 天前 评论：
16 天前 评论：
变更摘要
该 PR 是针对 v2.9.0 的构建 bugfix，解决 CANN 与 PTA（PyTorch Adapter）头文件混用导致的编译问题。核心思路是统一 ACL 头文件的引用方式与打包布局：将全部 #include "third_party/acl/inc/acl/xxx.h" 改为标准形式 #include <acl/xxx.h>，同时在 setup.py、build_libtorch_npu.py 中把 third_party/acl/inc 下的头文件按相对路径整体拷贝到安装目录 include/ 下（保留 acl/、graph/、profiling/、aml/ 等子目录结构），并同步调整测试脚本、build_stub.sh、cpp_builder.py、torch_npu/utils/cpp_extension.py 及 ci/access_control_test.py 中的 include 路径，使其直接使用安装后的 include 目录。
主要改动
- ACL 头文件引用方式统一：将约 60 个 C++/头文件中
#include "third_party/acl/inc/..."的写法（含acl.h、acl_base.h、acl_rt.h、acl_op_compiler.h、acl_prof.h、acl_mdl.h、super_kernel.h、error_codes/rt_error_codes.h、graph/operator.h、profiling/prof_api.h、aml/aml_fwk_detect.h等）统一改为#include <acl/...>等标准形式，避免与 CANN 自带头文件混用导致重复定义或版本不一致。 - 打包脚本头文件拷贝逻辑重构：
setup.py与build_libtorch_npu.py的copy_hpp/get_src_py_and_dst删除原先针对third_party/acl/inc/*/*.h的通配拷贝，改为用glob.glob(..., recursive=True)递归收集 ACL 头文件，并按相对路径拷入libtorch_npu/include/build/packages/torch_npu/include，保持acl/目录层级。 build_stub.sh支持自定义头文件目录：新增set -e与可选参数ACL_INCLUDE_DIR（默认../inc），各gcc编译命令改用${ACL_INCLUDE_DIR}作为-I路径；多个测试脚本（如test/allocator/test_pluggable_allocator_extensions.py、test/npu/test_allocator_trace_tracker.py、test/test_npu_expandable_segments.py、test/test_sanitizer_pluggable_allocator.py）随之把安装目录include传入该脚本，并移除include/third_party/acl/inc这一 include 路径。- 构建与工具链 include 路径调整：
torch_npu/_inductor/cpp_builder.py的include_paths与torch_npu/utils/cpp_extension.py的NpuExtension删除include/third_party/acl/inc路径，改用安装根目录include；ci/access_control_test.py的fetch_acl_headers回退拷贝路径由include/third_party/acl/inc/acl改为include/acl。 - 测试 C++ 用例与依赖头文件同步更新：
test/cpp_extensions/external_stream_test.cpp、pluggable_allocator_extensions.cpp及torch_npu/_inductor/ascend_npu_ir/npu/utils.py中的 ACL 头文件引用改为<acl/...>形式并移除旧 include 路径，third_party/hccl/inc/hccl/hccl.h内部对acl.h的引用同样改为<acl/acl.h>，保证头文件依赖链一致。
atomgit-bot
16 天前 评论：
16 天前 评论：
16 天前 添加了label：ascend-cla/yes
此处折叠了198条消息 查看更多
7 天前 添加了label：approved
7 天前 添加了label：lgtm
7 天前 合入了pull request
AtlasAccount
7 天前 评论：
7 天前 评论：
流水线 pytorch_gitcode_PR_multiVersion#14693 [ commitID：84a46bdd ] 已完成
【合入来源】
https://gitcode.com/Ascend/pytorch/issues/3991
【修改方案】
问题现象：
PTA 26.1.0 仓库中的 third_party/acl 头文件来自 CANN 9.1.0，而编译环境安装的是 CANN 9.0.0。当同一个编译单元同时包含两套版本的 ACL 头文件时，会出现类似以下错误：aclmdlRITask 未声明、未定义或类型不匹配
aclmdlRITask 是 CANN 9.1.0 头文件所依赖的定义，但 CANN 9.0.0 对应头文件中不存在该定义或定义不兼容。
根因分析：
1 当前头文件查找路径不统一：PTA 源码中大量使用以下写法
#include "third_party/acl/inc/acl/acl_mdl.h"
因为编译命令中包含 PTA 项目根目录，这种写法会直接命中 PTA 26.1.0 仓库内由 CANN 9.1.0 导入的头文件：torch_npu/third_party/acl/inc/acl/acl_mdl.h
但该头文件内部使用的是标准 SDK 相对路径：#include "acl/acl_base.h"
编译器无法相对 acl_mdl.h 所在目录找到 acl/acl_base.h，因此转而按照全部 -I 目录依次搜索。如果 CANN 9.0.0 的 include 路径排在 PTA 内置 ACL 路径前面，就会命中：CANN-9.0.0/include/acl/acl_base.h
同一个翻译单元最终形成：
acl_mdl.h -> PTA third_party 中的 CANN 9.1.0 版本
acl_base.h -> 环境 CANN 9.0.0 版本
这是一种“首层头文件由源码路径固定版本、传递头文件由 -I 顺序选择版本”的混合查找模式。
2 问题不局限于 acl_mdl.h
同样风险存在于所有能够继续 include 其他 CANN 头文件的入口，包括：
acl/...
aml/...
profiling/...
graph/...
ge/...
op_proto/...
因此不能只修复出现错误的某一个头文件，也不能只替换 acl_mdl.h。必须统一整个 CANN 头文件族的查找规则。
修改目标：
修改后需要满足
源码不再包含 third_party/acl/inc/... 这种仓库物理路径。
ACL、AML、profiling 等头文件全部通过统一 include root 查找。
一个编译 target 对 CANN 头文件只选择一套版本。
CANN 9.0.0 环境构建时，首层和传递头文件必须全部命中 CANN 9.0.0。
使用 PTA 内置头文件构建时，首层和传递头文件必须全部命中 PTA 内置的同一套版本。
wheel、libtorch_npu、C++ Extension 和 Inductor/AOT 使用同一目录契约
修改方案：
1、代码中使用 PTA 内置 ACL 头文件的地方，删除 third_party/acl/inc/ 物理路径前缀，统一改成从 include root 查找
2、打包或安装时，把 third_party/acl/inc 下需要公开的目录按原相对结构复制到安装产物的公共 include 根目录
【资料变更】
不涉及
【接口变更】
不涉及
【功能验证】
cann 9.0.0 + pta 26.1.0的vllm-ascend安装验证成功
【CheckList】