已合并
[fix] keep npu tensor strides for empty_like #31959
[fix] keep npu tensor strides for empty_like #31959
已合并
zhangqiongwen创建于 3月18日
zhangqiongwen成员
3月18日

【合入来源】

如有社区issue,请关联issue链接
请勿携带内部流程信息(需求链接、问题单、内部issue等)

【修改方案】

请描述修改内容的具体实现,涉及哪些组件之间进行交互,可以用1、2、3、...进行罗列
如果是需求或者重构类的PR,需要补充详细设计文档(说明上下游组件关系、时序图、类图、DFX能力等内容)

新增empty_stride分支,和原生对齐

【资料变更】

请确认是否涉及资料变更。如涉及,需要在PR中体现,并简要说明修改内容。如不涉及,需填写“不涉及”

【接口变更】

请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及,需要详细说明接口以及对应的变更内容,同时需要在资料中体现。如不涉及,需填写“不涉及”

【功能验证】

说明测试场景,测试方法。如果本次测试方式与常规单元测试不同,请详细说明您的测试步骤
新增/变更内容是否已新增/适配UT测试用例看护,并补充测试自验证截图

新增ut

【CheckList】

PR提交人对以下CheckList自检项进行全量自检,自检通过或不涉及,均修改 [ ] 为 [x]

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 zhangqiongwen 的贡献)
Zzhangqiongwen成员
3月18日 创建了 pull request,commit 996f2a3b
ascend-robot
ascend-robot成员
3月18日 评论:

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 htchu, chengpeng25 (2/2) htchu (1/1)
test htchu, chengpeng25 (2/2) htchu (1/1)

💡 Tip:

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

CLA Signature Pass

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

likedislike
ascend-robotascend-robot成员
3月18日 添加了label:ascend-cla/yes
zhangqiongwen成员
3月18日 评论:

compile

likedislike
AtlasAccount
AtlasAccount成员
3月18日 评论:

问题/功能描述

优化NPU设备上empty_like操作的实现逻辑,以提升内存布局的兼容性和性能。原实现在输入张量为基础格式、支持跨步操作且为连续布局时,新创建的张量未能继承原始步长,可能导致后续操作效率降低或兼容性问题。

修改方案描述

empty_like_npu函数中,当输入张量满足特定条件(NPU设备、基础格式、支持跨步、连续布局)时,改为通过at::infer_dense_strides推断步长并使用at::empty_strided创建新张量,从而保留原始内存布局特征。对于不满足条件的其他情况,则沿用原有的ApplyTensorWithFormat创建逻辑。

likedislike
ascend-robotascend-robot成员
3月18日 添加了label:ci-pipeline-running
ascend-robotascend-robot成员
3月18日 删除了label:ci-pipeline-running
ascend-robotascend-robot成员
3月18日 添加了label:ci-pipeline-failed
ascend-robot
ascend-robot成员
3月18日 评论:
流水线 PR-pipeline_pytorch#11249 运行失败
阶段 任务名 状态 详情
编译构建 Build_X86 >>>
Build_ARM >>>
Build_LibTorch_x86 >>>
Build_X86_torchair 🛑 >>>
Build_ARM_torchair 🛑 >>>
Build_ARM_inductor 🛑 >>>
恶意代码检查 Antipoison >>>
编码安全与规范检查 CodeCheck >>>
check_error >>>
开源片段检查 SCA >>>
开发者测试 UT_X86_Part_01 🛑 >>>
UT_X86_Part_02 🛑 >>>
UT_ARM_A3_Part_01 🛑 >>>
UT_ARM_A3_Part_02 🛑 >>>
UT_DIST_X86_Part_01 🛑 >>>
UT_DIST_X86_Part_02 🛑 >>>
UT_DIST_X86_Part_03 🛑 >>>
UT_DIST_X86_Part_04 🛑 >>>
UT_inductor_Part_01 🛑 >>>
UT_inductor_Part_02 🛑 >>>
UT_inductor_Part_03 🛑 >>>
UT_inductor_Part_04 🛑 >>>
UT_ARM_A2_Part_01 >>>
UT_ARM_A2_Part_02 >>>
流水线 PR-pipeline_pytorch >>>
likedislike
zhangqiongwen成员
3月19日 评论:

retry

likedislike
Zzhangqiongwen成员
3月24日 预合并成功(commit_id: f552cda098db736843dbea1eebf03f46e07964ff)
此处折叠了9条事件消息 查看更多
AtlasAccountAtlasAccount成员
3月24日 添加了label:ci-pipeline-failed
zhangqiongwen成员
3月24日 评论:

compile

likedislike
ascend-robotascend-robot成员
3月24日 删除了label:ci-pipeline-failed
ascend-robotascend-robot成员
3月24日 添加了label:ci-pipeline-running
ascend-robotascend-robot成员
3月24日 删除了label:ci-pipeline-running
ascend-robotascend-robot成员
3月24日 添加了label:ci-pipeline-failed
ascend-robot
ascend-robot成员
3月24日 评论:
流水线 PR-pipeline_pytorch#12184 运行失败
阶段 任务名 状态 详情
编译构建 Build_X86 🕚 >>>
Build_ARM 🕚 >>>
Build_LibTorch_x86 🕚 >>>
Build_X86_torchair 🕚 >>>
Build_ARM_torchair 🕚 >>>
Build_ARM_inductor 🕚 >>>
恶意代码检查 Antipoison >>>
编码安全与规范检查 CodeCheck >>>
check_error >>>
开源片段检查 SCA >>>
开发者测试 UT_X86_Part_01 🕚 >>>
UT_X86_Part_02 🕚 >>>
UT_ARM_A3_Part_01 🕚 >>>
UT_ARM_A3_Part_02 🕚 >>>
UT_DIST_X86_Part_01 🕚 >>>
UT_DIST_X86_Part_02 🕚 >>>
UT_DIST_X86_Part_03 🕚 >>>
UT_DIST_X86_Part_04 🕚 >>>
UT_inductor_Part_01 🕚 >>>
UT_inductor_Part_02 🕚 >>>
UT_inductor_Part_03 🕚 >>>
UT_inductor_Part_04 🕚 >>>
UT_ARM_A2_Part_01 🕚 >>>
UT_ARM_A2_Part_02 🕚 >>>
流水线 PR-pipeline_pytorch >>>
likedislike
zhangqiongwen成员
3月24日 评论:

compile

likedislike
ascend-robotascend-robot成员
3月24日 删除了label:ci-pipeline-failed
ascend-robotascend-robot成员
3月24日 添加了label:ci-pipeline-running
ascend-robotascend-robot成员
3月24日 删除了label:ci-pipeline-running
ascend-robotascend-robot成员
3月24日 添加了label:ci-pipeline-passed
ascend-robot
ascend-robot成员
3月24日 评论:
流水线 PR-pipeline_pytorch#12204 已完成
阶段 任务名 状态 详情
编译构建 Build_X86 >>>
Build_ARM >>>
Build_LibTorch_x86 >>>
Build_X86_torchair 🛑 >>>
Build_ARM_torchair 🛑 >>>
Build_ARM_inductor 🛑 >>>
恶意代码检查 Antipoison >>>
编码安全与规范检查 CodeCheck >>>
check_error >>>
开源片段检查 SCA >>>
开发者测试 UT_X86_Part_01 🛑 >>>
UT_X86_Part_02 🛑 >>>
UT_ARM_A3_Part_01 🛑 >>>
UT_ARM_A3_Part_02 🛑 >>>
UT_DIST_X86_Part_01 🛑 >>>
UT_DIST_X86_Part_02 🛑 >>>
UT_DIST_X86_Part_03 🛑 >>>
UT_DIST_X86_Part_04 🛑 >>>
UT_inductor_Part_01 🛑 >>>
UT_inductor_Part_02 🛑 >>>
UT_inductor_Part_03 🛑 >>>
UT_inductor_Part_04 🛑 >>>
UT_ARM_A2_Part_01 >>>
UT_ARM_A2_Part_02 >>>
流水线 PR-pipeline_pytorch >>>
likedislike
Zzhangqiongwen成员
3月25日 修改了pull request 的描述
htchu成员
3月25日 评论:

/approve

likedislike
ascend-robotascend-robot成员
3月25日 添加了label:approved
chengpeng25成员
3月25日 评论:

/lgtm

likedislike
ascend-robotascend-robot成员
3月25日 添加了label:lgtm
ascend-robot
ascend-robot成员
3月25日 评论:

Review Guide

This pull-request passes review.
Committers who wrote a comment of /approve are: htchu.
Reviewers who wrote a comment of /lgtm are: htchu, chengpeng25.

likedislike
ascend-robotascend-robot成员
3月25日 合入了pull request