已关闭
test(fx): add ShapeEnv API tests without upstream coverage #36447
小辉懂编程创建于 5月22日关闭于 5月22日
test(fx): add ShapeEnv API tests without upstream coverage #36447
已关闭
当前Pull Request已关闭, 关闭人@小辉懂编程
ascend-robot
5月22日 评论：
5月22日 评论：
ascend-robot
5月22日 评论：
5月22日 评论：
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
⚠️ This PR does not yet meet the following requirements:lgtm (requires ≥ 2 person(s) per module)、approve (requires ≥ 1 person(s) per module)
Module Approval Details
| module | lgtm status | approve status |
|---|---|---|
| test | ❌ (0/2)(You can also ask: medivh-x, kisnwang, wangmin0104, 褚博宁, chengpeng25) | ❌ (0/1)(You can also ask: liujunzhu, crazyDannyBoy, wangmin0104, 王朝, suhaibo) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CLA Signature Pass
gehui-2026, thanks for your pull request. All authors of the commits have signed the CLA. 👍
5月22日 添加了label：ascend-cla/yes
ascend-robot
5月22日 评论：
5月22日 评论：
当前仓库存在以下 保护分支 ：
| Protected Branch | Version | Release |
|---|---|---|
| master | ||
| v2.10.0 | ||
| v2.7.1 | ||
| v2.9.0 | ||
| v2.11.0 | ||
| v2.12.0 |
评论 /sync <branch1> <branch2> ... 可将当前 PR 修改同步到其它分支（创建同步 PR）：
a) 如果当前 PR 是 Open 状态，同步操作将延迟到 PR 被合并时执行
b) 如果当前 PR 已经 Merged，将立即执行同步操作
注意：
- /sync 命令可以指定同步到多个分支，仅最后一个 /sync 命令生效
- 如果创建的同步 PR 不正确，可通过向同步 PR 的源分支提交轻量级 PR 完善，或使用 /close 命令关闭
小辉懂编程
5月22日 评论：
5月22日 评论：
compile
5月22日 添加了label：ci-pipeline-running
ascend-robot
5月22日 评论：
5月22日 评论：
Ascend docs pipeline is running...
5月22日 添加了label：docs-ci-pipeline-running
ascend-robot
5月22日 评论：
5月22日 评论：
Ascend docs pipeline is running...
ascend-robot
5月22日 评论：
5月22日 评论：
✅ 跳过 docs ci 检查，没有需要检查的文档文件
5月22日 删除了label：docs-ci-pipeline-running
5月22日 添加了label：docs-ci-pipeline-success
ascend-robot
5月22日 评论：
5月22日 评论：
✅ 跳过 docs ci 检查，没有需要检查的文档文件
5月22日 预合并成功（commit_id: 91d260dba108a485d25aff49ed92006865a8a385）
此处折叠了5条事件消息 查看更多
5月22日 删除了label：ci-pipeline-running
ascend-robot
5月22日 评论：
5月22日 评论：
流水线 PR-pipeline_pytorch#25039 已终止运行
| 阶段 | 任务名 | 状态 | 详情 |
|---|---|---|---|
| 编译构建 | 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_X86_Part_01 | ✅ | >>> |
| UT_X86_Part_02 | ✅ | >>> | |
| 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 | 🛑 | >>> | |
| 流水线 | PR-pipeline_pytorch | 🟨 | >>> |
- compile : 运行流水线
- retry : 重试流水线所有失败子任务
- retry <任务名> : 仅重试指定失败子任务
- stop : 停止流水线
ascend-robot
5月22日 评论：
5月22日 评论：
Ascend docs pipeline is running...
5月22日 删除了label：docs-ci-pipeline-success
5月22日 添加了label：docs-ci-pipeline-running
5月22日 修改了pull request 的描述
ascend-robot
5月22日 评论：
5月22日 评论：
✅ 跳过 docs ci 检查，没有需要检查的文档文件
5月22日 删除了label：docs-ci-pipeline-running
5月22日 添加了label：docs-ci-pipeline-success
小辉懂编程
5月22日 评论：
5月22日 评论：
compile
5月22日 添加了label：ci-pipeline-running
ascend-robot
5月22日 评论：
5月22日 评论：
Ascend docs pipeline is running...
5月22日 删除了label：docs-ci-pipeline-success
5月22日 添加了label：docs-ci-pipeline-running
ascend-robot
5月22日 评论：
5月22日 评论：
✅ 跳过 docs ci 检查，没有需要检查的文档文件
5月22日 删除了label：docs-ci-pipeline-running
5月22日 添加了label：docs-ci-pipeline-success
5月22日 删除了label：ci-pipeline-running
5月22日 添加了label：ci-pipeline-passed
ascend-robot
5月22日 评论：
5月22日 评论：
流水线 PR-pipeline_pytorch#25057 [ commitID：c25c24b6 ] 已完成
| 阶段 | 任务名 | 状态 | 详情 |
|---|---|---|---|
| 编译构建 | 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_X86_Part_01 | ✅ | >>> |
| UT_X86_Part_02 | ✅ | >>> | |
| 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 | 🛑 | >>> | |
| 流水线 | PR-pipeline_pytorch | ✅ | >>> |
- compile : 运行流水线
- retry : 重试流水线所有失败子任务
- retry <任务名> : 仅重试指定失败子任务
- stop : 停止流水线
棕
5月22日 评论：
5月22日 评论：
- pr描述中最上面需要贴一下你的 任务 issue 链接。
- 多个提交需要 squash 成一个。
- 上游社区具有测试用例的两个API，你是否验证其能跑通？需要在 PR 描述中提供相关证明，证明上游社区针对这两个API 的测试，覆盖全面无误。
- 测试用例，如果你在原有的.py 上面改，还是要按照我们提供的文档规范来，包括开头的注释说明，以及你最好创建个测试用例的类，集成你的测试方法。
- 需要提供你的测试用例跑通的结果证明。
5月22日 关闭了 pull request
【合入来源】
Torch-NPU ShapeEnv API 补齐任务。
本 PR 为 v2.7.1 版本的自写测试 PR，覆盖以下 ShapeEnv API：
torch.fx.experimental.symbolic_shapes.ShapeEnv.get_pruned_guardstorch.fx.experimental.symbolic_shapes.ShapeEnv.is_unbacked_syminttorch.fx.experimental.symbolic_shapes.ShapeEnv.ignore_fresh_unbacked_symbols这 3 个 API 在 PyTorch 官方社区目前没有直接的测试用例覆盖，因此需要在 Torch-NPU 中新增自写测试。
本 PR 属于：
【修改方案】
本 PR 修改文件：
具体修改内容如下：
import sympy导入。test_shape_env_get_pruned_guards测试函数，测试ShapeEnv.get_pruned_guards的基本功能。test_shape_env_is_unbacked_symint测试函数，测试ShapeEnv.is_unbacked_symint的基本功能。test_shape_env_ignore_fresh_unbacked_symbols测试函数，测试ShapeEnv.ignore_fresh_unbacked_symbols的基本功能。一、API 功能
1.
torch.fx.experimental.symbolic_shapes.ShapeEnv.get_pruned_guards该 API 用于获取已剪枝的 guards 列表。在 symbolic tracing 过程中，当某些符号约束被证明为平凡条件或不再需要时，
ShapeEnv会将其从 guards 集合中移除。get_pruned_guards返回这些已被移除的 guards，用于调试和验证符号形状化简逻辑。2.
torch.fx.experimental.symbolic_shapes.ShapeEnv.is_unbacked_symint该 API 用于判断给定的 sympy 符号是否为 unbacked symint。Unbacked symint 是指没有具体 backing value 的符号整数，通常在 symbolic tracing 过程中动态创建。该 API 用于区分 unbacked 符号和普通符号，是符号形状管理的重要辅助函数。
3.
torch.fx.experimental.symbolic_shapes.ShapeEnv.ignore_fresh_unbacked_symbols该 API 提供一个上下文管理器，用于在 symbolic tracing 过程中忽略新创建的 unbacked 符号。在某些场景下，临时创建的 unbacked 符号不需要被追踪，使用此 API 可以避免不必要的符号约束。
二、官方测试用例完整性分析
PyTorch 官方社区目前没有直接针对这 3 个 API 的测试用例。
因此，本 PR 新增自写测试，覆盖这 3 个 API 的基本功能。
三、NPU 适配
这 3 个 API 属于 ShapeEnv 的 Python 层符号形状管理逻辑，不涉及 NPU 设备侧行为。
因此，本 PR 的测试代码不涉及 NPU 设备检测，仅测试 Python 层 API 的基本功能。
【资料变更】
不涉及。
本 PR 仅新增测试代码，不修改文档资料。
【接口变更】
不涉及。
本 PR 不新增、不删除、不修改 Python API 或 C++ 接口，仅新增测试代码。
【功能验证】
测试场景
test_shape_env_get_pruned_guards：创建ShapeEnv实例，生成 unbacked symint，调用get_pruned_guards并验证返回类型为 list。test_shape_env_is_unbacked_symint：创建ShapeEnv实例，生成 unbacked symint 和普通 sympy 符号，调用is_unbacked_symint并验证对两者的判断结果。test_shape_env_ignore_fresh_unbacked_symbols：创建ShapeEnv实例，使用ignore_fresh_unbacked_symbols上下文管理器，在其中创建 unbacked symint 并验证创建成功。测试方法
本 PR 的测试代码使用
unittest.TestCase框架，通过self.assertIsInstance、self.assertTrue、self.assertFalse等断言方法验证 API 行为。测试自验证
由于 Web IDE 环境缺少已编译的
torch_npu._C扩展，本地测试执行会在import torch_npu阶段失败。这是环境限制，不是代码错误。实际测试验证依赖 CI 流水线。【CheckList】
关联 PR
v2.12.0 同类自写测试 PR：
v2.11.0 同类自写测试 PR：
v2.10.0 同类自写测试 PR：
v2.9.0 同类自写测试 PR：
master 同类自写测试 PR：