已合并
pta supports to create nested tensor #30227
culechan创建于 1月29日
pta supports to create nested tensor #30227
已合并
Pull Request已成功合入, 合并人@ascend-robot
(感谢 culechan 的贡献)AtlasAccount
1月29日 评论:
1月29日 评论:
ascend-robot
1月29日 评论:
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 |


1月29日 添加了label:ascend-cla/yes
ascend-robot
1月29日 评论:
1月29日 评论:
culechan
1月29日 评论:
1月29日 评论:
compile


AtlasAccount
1月29日 评论:
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的全面参数化测试,覆盖了不同维度、数据类型和功能方法,以保障新策略下的功能质量。


1月29日 添加了label:ci-pipeline-running
1月29日 删除了label:ci-pipeline-running
1月29日 添加了label:ci-pipeline-passed
ascend-robot
1月29日 评论:
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 | ✅ | >>> |


zhangqiongwen
1月30日 评论:
1月30日 评论:
/approve


1月30日 添加了label:lgtm
1月30日 添加了label:approved
ascend-robot
1月30日 评论:
1月30日 评论:
1月30日 合入了pull request
What type of PR is this?
What does this PR do / why do we need it:
Which issue(s) this PR fixes:
Fixes #
Special notes for your reviewers: