Pull Request

接口评审时间：2026/08/06(周四)

描述

PCIE Through 是 Ascend NPU 硬件能力，允许 device 端算子通过 PCIE 总线直接访问 host 内存。当前单算子场景可用，但在模型编译场景下，编译期无法确定输入 tensor 的真实的地址范围，导致 tiling 阶段无法选择正确 kernel。

本特性通过编译期能力识别 + TilingContext 注入 + 执行期上下文扩展（动态 shape），解决此问题。

核心设计

编译阶段（lowering）：

• BuildTilingPcieThroughInput 从 DefaultOpImplSpaceRegistryV2 查询算子是否支持 PCIe Through，同时通过 IsPcieThroughEnabled 检查运行环境（option 开关、芯片能力、设备连接类型），两者均满足时将 pcie_through_flag 初始值设为 true
• IsPcieThroughEnabled 结果通过 thread_local static 缓存，每个线程仅计算一次
• BuildTilingCommonInputs 在 input_addrs 可用时，通过 BuildRefTensor 将输入从 StorageShape 升级为包含设备地址的 Tensor，使运行时能通过 GetInputTensor(i)->GetAddr() 获取实际输入地址

运行时（executor）：

• UpdatePcieThroughFlag 读取 pcie_through_flag，为 false 时直接跳过；为 true 时通过 IsAnyInputInPcieRange 遍历输入地址，检查是否落在 PCIe 地址范围内
• 若存在 PCIe 地址 → flag 保持 true；否则改为 false
• GetPcieThroughFlag() 接口供 Tiling 函数和缓存键计算使用

关键修复：

• fwk_data_idx 恢复后向计算（input_num - kFwkDataOffset），修复 symbol tiling 路径中前向计算导致的 702 行 ASSERT 失败
• GetPcieThroughFlag 改为后向 GetInputNum() - 1，与 pcie_through_flag 作为最后一个输入的布局一致
• TilingForAtomic 补充 BuildTilingPcieThroughInput 调用，保证输入布局一致性
• BuildGeneralTilingCacheKey 简化为直接读 GetPcieThroughFlag()，消除运行时查 space_registry 的版本不一致问题

变更类型

• ✨ 新功能
• 🐛 Bug 修复

关联的Issue

如何测试

1. 编译 UT：make -C cmake-build-gcov ut_fast_runtime2_test ut_exe_graph ut_register -j$(nproc)
2. 运行 PCIe Through 相关 UT：--gtest_filter='*PcieThrough*'
3. 运行 TilingContextBuilder UT：--gtest_filter='TilingContextBuilderUT.*'
4. 在 PCIe 直通环境上执行模型推理，验证 pcie_through 算子正确识别地址并优化搬运

核对清单

• 我的代码遵循了项目的代码风格
• 我已对代码进行了自测
• 我已更新了相关的文档
• 我在标题中使用了合适的类型标签
• 我已经详细阅读了贡献指南（CONTRIBUTING.md），并遵守了其中的所有规定

其他信息

涉及文件