已合并
pta supports to create nested tensor #30227
pta supports to create nested tensor #30227
已合并
culechan创建于 1月29日
culechan
culechan成员
1月29日

What type of PR is this?

Uncomment only one /kind <> line, hit enter to put that in a new line, and remove leading whitespaces from that line:

/kind bug
/kind task
/kind feature

What does this PR do / why do we need it:

Which issue(s) this PR fixes:

Fixes #

Special notes for your reviewers:

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 culechan 的贡献)
culechanculechan成员
1月29日 创建了 pull request,commit 70320856
ascend-robot
ascend-robot成员
1月29日 评论:

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

以下是根据您提交的修改文件推荐的Reviewer和Committer序列,需各模块评审通过后方可合入.

Module List Reviewers Committers
repo-Ascend/pytorch wangchao285, guoqi1024, shaoyf, zhaozhijie, wangchao430 wasd1111222, wangqiang160, medivh-x, zqwenn, li_jing_hw
likedislike
ascend-robotascend-robot成员
1月29日 添加了label:ascend-cla/yes
ascend-robot
ascend-robot成员
1月29日 评论:

CLA Signature Pass

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

likedislike
culechan
culechan成员
1月29日 评论:

compile

likedislike
AtlasAccount
AtlasAccount成员
1月29日 评论:

问题/功能描述

本次PR调整了NPU设备对PyTorch嵌套张量(Nested Tensor)API的支持策略。核心目标是移除原有的运行时拦截和错误抛出机制,转而允许相关API在NPU上运行,并通过新增独立的测试文件来验证其功能正确性,从而将支持策略从“明确禁止”转变为“允许并测试”。

修改方案描述

修改方案主要包括移除旧逻辑和新增测试两部分。首先,移除了npu_intercept.py中用于拦截嵌套张量API的检查函数及其应用代码,并清理了unsupport_api.py中相关的定义。其次,删除了旧测试文件中验证运行时错误的四个测试用例。最后,新增了独立的test_nestedtensor.py测试文件,其中包含对torch.nested.nested_tensor API的全面参数化测试,覆盖了不同维度、数据类型和功能方法,以保障新策略下的功能质量。

likedislike
ascend-robotascend-robot成员
1月29日 添加了label:ci-pipeline-running
ascend-robotascend-robot成员
1月29日 删除了label:ci-pipeline-running
ascend-robotascend-robot成员
1月29日 添加了label:ci-pipeline-passed
ascend-robot
ascend-robot成员
1月29日 评论:
流水线 PR-pipeline_pytorch#5260 已完成
阶段 任务名 状态 详情
编译构建 Build_X86 ✅ >>>
Build_LibTorch ✅ >>>
Build_ARM ✅ >>>
Build_ARM_inductor 🛑 >>>
Build_X86_py311 🛑 >>>
Build_ARM_py311 🛑 >>>
dist_test_or_not ✅ >>>
恶意代码检查 Antipoison ✅ >>>
编码安全与规范检查 CodeCheck ✅ >>>
check_error ✅ >>>
开源片段检查 SCA ✅ >>>
开发者测试 UT_DIST_X86 🛑 >>>
UT_X86_Part_01 ✅ >>>
UT_X86_Part_02 ✅ >>>
UT_ARM_A2_Part_01 🛑 >>>
UT_ARM_A2_Part_02 🛑 >>>
UT_inductor_arm 🛑 >>>
流水线 PR-pipeline_pytorch ✅ >>>
likedislike
zhangqiongwen成员
1月30日 评论:

/approve

likedislike
ascend-robotascend-robot成员
1月30日 添加了label:lgtm
shaoyf成员
1月30日 评论:

/approve

likedislike
ascend-robotascend-robot成员
1月30日 添加了label:approved
ascend-robot
ascend-robot成员
1月30日 评论:

Review Guide

This Pull-Request Passes Review.
Committers who wrote a comment of /approve are: zqwenn, shaoyf.
Reviewers who wrote a comment of /lgtm are: zqwenn, shaoyf.

likedislike
ascend-robotascend-robot成员
1月30日 合入了pull request