已合并
pta supports to create nested tensor #30195
culechan创建于 1月29日
pta supports to create nested tensor #30195
已合并
Pull Request已成功合入, 合并人@ascend-robot
(感谢 culechan 的贡献)ascend-robot
1月29日 评论:
1月29日 评论:
AtlasAccount
1月29日 评论:
1月29日 评论:
ascend-robot
1月29日 评论:
1月29日 评论:
以下是根据您提交的修改文件推荐的Reviewer和Committer序列,需各模块评审通过后方可合入
| Module List | Reviewers | Committers |
|---|---|---|
| repo-Ascend/pytorch | guoguanghao, yuhaiyan, culechan, qinchuanyu, huangyunlong2022 | zqwenn, li_jing_hw, chenhao_1209, wangchao430, wangchao285 |


1月29日 添加了label:ascend-cla/yes
ascend-robot
1月29日 评论:
1月29日 评论:
AtlasAccount
1月29日 评论:
1月29日 评论:
问题/功能描述
本次PR旨在支持NPU设备上的嵌套张量(Nested Tensor)操作。主要解决了先前版本中,在NPU设备上错误地将torch.nested.nested_tensor和torch.nested.as_nested_tensor等API标记为不支持,并抛出运行时拦截的问题,从而恢复了这些API在NPU上的正常功能。
修改方案描述
修改方案包含两部分核心内容:一是从NPU设备的运行时拦截和错误处理机制中移除了嵌套张量相关API,删除了对应的拦截逻辑、检查函数及API黑名单,使其能够被正常调用。二是新增了一个独立的测试文件,用于全面验证嵌套张量在NPU环境下的功能,包括2D/3D张量创建、解绑、属性获取及尺寸检查等关键用例。


culechan
1月29日 评论:
1月29日 评论:
compile


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#5170 已完成
| 阶段 | 任务名 | 状态 | 详情 |
|---|---|---|---|
| 编译构建 | 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 | ✅ | >>> |


1月29日 添加了label:lgtm
1月29日 添加了label:approved
ascend-robot
1月29日 评论:
1月29日 评论:
Review Guide
This Pull-Request Passes Review.
Committers who wrote a comment of /approve are: wangchao430, shaoyf.
Reviewers who wrote a comment of /lgtm are: wangchao430, shaoyf.


1月29日 合入了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: