Pull Request已成功合入, 合并人@ascend-robot
(感谢 21xiu 的贡献)变更摘要
此 PR 在 FormatCastKernelNpu.cpp 的 MaybeUseAclnnNpuFormatCast 函数中新增了一个防护判断:当对形状与存储形状不一致的 2D 连续张量执行 ND → FRACTAL_NZ 格式转换时(例如 [1,N] 经 transpose() 后变为 [N,1] 的列向量),回退到非 aclnn 路径,以规避下游算子的精度问题。
主要改动
- 新增形状与存储形状一致性检查:在
MaybeUseAclnnNpuFormatCast中,当源格式为ACL_FORMAT_ND、目标格式为ACL_FORMAT_FRACTAL_NZ、张量为 2D 连续且最后一维大小为 1 时,比对src.sizes()与src_desc.base_sizes_,若两者不一致则判定为 view 操作(如 transpose),返回false回退到非 aclnn 路径 - 保护范围限定为 N×1 列向量场景:条件中通过
src.sizes()[1] == 1将防护范围限定为 reshape 后的 N×1 列向量,避免对其他形状的 ND→NZ 转换造成不必要的影响 - 无删除、仅新增:此改动为纯增量修改,共 12 行新增代码,不影响现有逻辑路径


代码审查
经过对整个变更的深入分析,包括上下文代码、调用链、数据结构定义以及边界条件的完整审查,我未发现该 diff 引入的任何逻辑错误、安全风险、可靠性问题或性能退化。
变更内容是在 MaybeUseAclnnNpuFormatCast 函数中新增一个守卫条件:当对 shape 为 [N, 1] 的 2D 连续 tensor 做 ND→FRACTAL_NZ 格式转换时,若其 storage 的 base_sizes_ 与当前 sizes() 不一致(即为 view/transpose 产生),则回退到非 aclnn 路径,以避免精度问题。
审查结论:未发现需要报告的问题。
各文件审查结果
torch_npu/csrc/aten/common/FormatCastKernelNpu.cpp:+12 行,无问题。
Closing Summary
- 发现数量(按严重性):P0: 0, P1: 0, P2: 0, P3: 0
- 整体风险判断:低风险。该变更是针对性的防御性回退,逻辑正确,条件检查完整(包含 size 边界保护),与现有代码风格一致,调用方对回退路径的处理亦正确。
⚠️ 已识别出整体风险,但无法提取行内评论,请参考整体评估。


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 |
|---|---|---|
| repo-Ascend/pytorch | ✅ 楚浩田, chengpeng25 (2/2) | ✅ 楚浩田 (1/1) |
| test | ✅ 楚浩田, chengpeng25 (2/2) | ✅ 楚浩田 (1/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
xiu_21, thanks for your pull request. All authors of the commits have signed the CLA. 👍


Linking Issue Notice
@xiu_21 , the pull request must be linked to at least one issue.
If an issue has already been linked, but the needs-issue label remains, you can remove the label by commenting /check-issue .


当前仓库存在以下 保护分支 :
| Protected Branch | Version | Release |
|---|---|---|
| master | ||
| v2.9.0 | ||
| v2.12.0 | ||
| v2.11.0 | ||
| v2.7.1 | ||
| v2.10.0 | ||
| v2.7.1-26.1.0 | ||
| v2.9.0-26.1.0 | ||
| v2.12.0-26.1.0 | ||
| v2.10.0-26.1.0 | ||
| v2.11.0-26.1.0 | ||
| ci-test |
评论 /sync <branch1> <branch2> ... 可将当前 PR 修改同步到其它分支(创建同步 PR):
a) 如果当前 PR 是 Open 状态,同步操作将延迟到 PR 被合并时执行
b) 如果当前 PR 已经 Merged,将立即执行同步操作
注意:
- /sync 命令可以指定同步到多个分支,仅最后一个 /sync 命令生效
- 如果创建的同步 PR 不正确,可通过向同步 PR 的源分支提交轻量级 PR 完善,或使用 /close 命令关闭


ascend docs pipeline is running...


✅ 跳过 docs ci 检查,没有需要检查的文档文件


| 阶段 | 任务名 | 状态 | 详情 |
|---|---|---|---|
| 编译构建 | Build_X86 | ✅ | >>> |
| Build_ARM | ✅ | >>> | |
| Build_LibTorch_x86 | ✅ | >>> | |
| Build_LibTorch_ARM | ✅ | >>> | |
| Build_X86_torchair | 🛑 | >>> | |
| Build_ARM_torchair | 🛑 | >>> | |
| patch_test | 🛑 | >>> | |
| 恶意代码检查 | Antipoison | ✅ | >>> |
| 编码安全与规范检查 | CodeCheck | ✅ | >>> |
| check_error | ✅ | >>> | |
| CodeCheck_lintrunner | ✅ | >>> | |
| 开源片段检查 | SCA | ✅ | >>> |
| 开发者测试 | UT_ARM_A3_Part_01 | 🛑 | >>> |
| UT_ARM_A3_Part_02 | 🛑 | >>> | |
| UT_ARM_A2_Part_01 | ✅ | >>> | |
| UT_ARM_A2_Part_02 | ✅ | >>> | |
| UT_ARM_A2_Part_03 | ✅ | >>> | |
| UT_inductor_Part_01 | 🛑 | >>> | |
| UT_inductor_Part_02 | 🛑 | >>> | |
| UT_inductor_Part_03 | 🛑 | >>> | |
| UT_inductor_Part_04 | 🛑 | >>> | |
| UT_DIST_ARM_Part_01 | 🛑 | >>> | |
| UT_DIST_ARM_Part_02 | 🛑 | >>> | |
| UT_DIST_ARM_Part_03 | 🛑 | >>> | |
| UT_DIST_ARM_Part_04 | 🛑 | >>> | |
| UT_ARM_A2_Select_Part_01 | ✅ | >>> | |
| UT_ARM_A2_Select_Part_02 | ✅ | >>> | |
| 流水线 | PR-pipeline_pytorch | ✅ | >>> |
- compile、compile_inductor、compile_torchair : 运行流水线
- retry : 重试流水线所有失败子任务
- retry <任务名> : 仅重试指定失败子任务
- stop : 停止流水线


compile


ascend docs pipeline is running...


ascend docs pipeline is running...


✅ 跳过 docs ci 检查,没有需要检查的文档文件


✅ 跳过 docs ci 检查,没有需要检查的文档文件


| 阶段 | 任务名 | 状态 | 详情 |
|---|---|---|---|
| 编译构建 | Build_X86 | ✅ | >>> |
| Build_ARM | ✅ | >>> | |
| Build_LibTorch_x86 | ✅ | >>> | |
| Build_LibTorch_ARM | ✅ | >>> | |
| Build_X86_torchair | 🛑 | >>> | |
| Build_ARM_torchair | 🛑 | >>> | |
| patch_test | 🛑 | >>> | |
| 恶意代码检查 | Antipoison | ✅ | >>> |
| 编码安全与规范检查 | CodeCheck | ✅ | >>> |
| check_error | ✅ | >>> | |
| CodeCheck_lintrunner | ✅ | >>> | |
| 开源片段检查 | SCA | ✅ | >>> |
| 开发者测试 | UT_ARM_A3_Part_01 | 🛑 | >>> |
| UT_ARM_A3_Part_02 | 🛑 | >>> | |
| UT_ARM_A2_Part_01 | ✅ | >>> | |
| UT_ARM_A2_Part_02 | ✅ | >>> | |
| UT_ARM_A2_Part_03 | ✅ | >>> | |
| UT_inductor_Part_01 | 🛑 | >>> | |
| UT_inductor_Part_02 | 🛑 | >>> | |
| UT_inductor_Part_03 | 🛑 | >>> | |
| UT_inductor_Part_04 | 🛑 | >>> | |
| UT_DIST_ARM_Part_01 | 🛑 | >>> | |
| UT_DIST_ARM_Part_02 | 🛑 | >>> | |
| UT_DIST_ARM_Part_03 | 🛑 | >>> | |
| UT_DIST_ARM_Part_04 | 🛑 | >>> | |
| UT_ARM_A2_Select_Part_01 | ✅ | >>> | |
| UT_ARM_A2_Select_Part_02 | ✅ | >>> | |
| 流水线 | PR-pipeline_pytorch | ✅ | >>> |
- compile、compile_inductor、compile_torchair : 运行流水线
- retry : 重试流水线所有失败子任务
- retry <任务名> : 仅重试指定失败子任务
- stop : 停止流水线


/lgtm




【合入来源】
【修改方案】
shape为[1,n]的tensor做transpose()后再做ND转NZ当前有精度问题,回退至aclop
【资料变更】
不涉及
【接口变更】
不涉及
【功能验证】
新增用例pytest test/custom_ops/test_npu_format_cast.py -k test_2d_view_nx1_nd_to_nz_format_id


修改前:
修改后:
【CheckList】