已合并

test: add ShapeEnv API alignment tests for symbolic_shapes #38160

liuhaodong-2026创建于 6月10日从liuhaodong-2026/pytorch： test-shape-env-api-alignment-v2.10.0 合入到 Ascend/pytorch： v2.10.0

test: add ShapeEnv API alignment tests for symbolic_shapes #38160

已合并

liuhaodong-2026创建于 6月10日

从liuhaodong-2026/pytorch test-shape-env-api-alignment-v2.10.0合入到Ascend/pytorch v2.10.0

liuhaodong-2026

6月10日

【合入来源】

issue/工单

关联 issue: #1624

【修改方案】

为 torch.fx.experimental.symbolic_shapes.ShapeEnv 的 5 个 API 补齐测试用例：

新增测试文件 test/fx/test_shape_env_apis.py，覆盖以下 API：
- torch.fx.experimental.symbolic_shapes.ShapeEnv.deserialize_symexpr
- torch.fx.experimental.symbolic_shapes.ShapeEnv.evaluate_symexpr
- torch.fx.experimental.symbolic_shapes.ShapeEnv.evaluate_guards_expression
- torch.fx.experimental.symbolic_shapes.ShapeEnv.evaluate_guards_for_args
- torch.fx.experimental.symbolic_shapes.ShapeEnv.evaluate_sym_node
用例完备性分析：
- ShapeEnv.evaluate_guards_expression：PyTorch 社区已有相关用例，位于 test/test_dynamic_shapes.py，主要为符号表达式验证，不涉及 NPU 张量。本 PR 补充直接 API 调用验证。
- ShapeEnv.evaluate_sym_node：PyTorch 社区已有相关用例，位于 test/dynamo/test_dynamic_spec.py，主要为错误行为验证。本 PR 补充正常路径 API 调用验证。
- ShapeEnv.deserialize_symexpr、ShapeEnv.evaluate_symexpr、ShapeEnv.evaluate_guards_for_args：PyTorch 社区无直接测试用例，本 PR 补充对应直接调用测试。
测试场景覆盖：
- 常量符号表达式解析与计算。
- 带符号变量的表达式解析。
- guard 表达式返回值验证。
- FakeTensorMode 下 placeholder tensor 与真实 NPU tensor 的 guard 校验。
- SymNode 在不同 size_oblivious 参数下的求值验证。

【资料变更】

不涉及（文档已在独立 PR #38301 中提交）

【接口变更】

不涉及

【功能验证】

在 NPU 环境下执行新增测试文件：

python test/fx/test_shape_env_apis.py

执行结果：

Ran 14 tests in 1.119s OK

【CheckList】

代码注释完备，正确记录错误日志
代码实现进行了返回值、空指针等校验
PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等
PR持续集成流水线（CI）执行通过，代码检查无异常

Pull Request已成功合入, 合并人@ascend-robot

（感谢 liuhaodong-2026 的贡献）

L liuhaodong-2026

6月10日创建了 pull request，commit a1de29fc

L liuhaodong-2026

6月10日关联了issue：【开源实习】Torch-NPU API一致性对齐：补齐测试用例、API功能对齐、补齐文档（18）

ascend-robot成员

6月10日评论：

openLiBing可接管PR, PR详情链接:
https://www.openlibing.com/apps/transitionPrDetail?owner=Ascend&repo=pytorch&number=38160&platform=gitcode&codeHostingPlatformFlag=gitcode

ascend-robot成员

6月10日添加了label：ascend-cla/yes

ascend-robot成员

6月10日评论：

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
test	✅ suhaibo, huangjingwei (2/2)	✅ suhaibo (1/1)

💡 Tip:

Committer can comment /approve or /lgtm

Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

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

ascend-robot成员

6月10日评论：

当前仓库存在以下 保护分支 ：

Protected Branch	Version	Release
master
v2.9.0
v2.11.0
v2.10.0
v2.12.0
v2.7.1
v2.7.1-26.1.0
v2.9.0-26.1.0
v2.12.0-26.1.0
v2.11.0-26.1.0
v2.10.0-26.1.0
ci-test
sync-pr28113--to-v2.9.0

评论 /sync <branch1> <branch2> ... 可将当前 PR 修改同步到其它分支（创建同步 PR）：
a) 如果当前 PR 是 Open 状态，同步操作将延迟到 PR 被合并时执行
b) 如果当前 PR 已经 Merged，将立即执行同步操作

注意：

/sync 命令可以指定同步到多个分支，仅最后一个 /sync 命令生效

如果创建的同步 PR 不正确，可通过向同步 PR 的源分支提交轻量级 PR 完善，或使用 /close 命令关闭

ascend-robot成员

6月10日评论：

Ascend docs pipeline is running...

ascend-robot成员

6月10日添加了label：docs-ci-pipeline-running

ascend-robot成员

6月10日评论：

✅ 跳过 docs ci 检查，没有需要检查的文档文件

ascend-robot成员

6月10日删除了label：docs-ci-pipeline-running

ascend-robot成员

6月10日添加了label：docs-ci-pipeline-success

L liuhaodong-2026

6月11日修改了pull request 的描述

L liuhaodong-2026

6月16日修改了pull request 的描述

Jingwei Huang成员

6月18日评论：

/lgtm

liuhaodong-2026

6月18日评论：

compile

ascend-robot成员

6月18日添加了label：ci-pipeline-running

ascend-robot成员

6月18日评论：

Ascend docs pipeline is running...

ascend-robot成员

6月18日删除了label：docs-ci-pipeline-success

ascend-robot成员

6月18日添加了label：docs-ci-pipeline-running

ascend-robot成员

6月18日评论：

✅ 跳过 docs ci 检查，没有需要检查的文档文件

ascend-robot成员

6月18日删除了label：docs-ci-pipeline-running

ascend-robot成员

6月18日添加了label：docs-ci-pipeline-success

ascend-robot成员

6月19日删除了label：ci-pipeline-running

ascend-robot成员

6月19日添加了label：ci-pipeline-passed

ascend-robot成员

6月19日评论：

流水线 PR-pipeline_pytorch#33552 已完成

阶段	任务名	状态	详情
编译构建	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	🛑	>>>
	UT_ARM_A2_Select_Part_01	✅	>>>
	UT_ARM_A2_Select_Part_02	✅	>>>
流水线	PR-pipeline_pytorch	✅	>>>

liuhaodong-2026

6月29日评论：

compile

ascend-robot成员

6月29日删除了label：ci-pipeline-passed

ascend-robot成员

6月29日添加了label：ci-pipeline-running

ascend-robot成员

6月29日评论：

ascend docs pipeline is running...

ascend-robot成员

6月29日删除了label：docs-ci-pipeline-success

ascend-robot成员

6月29日添加了label：docs-ci-pipeline-running

ascend-robot成员

6月29日评论：

✅ 跳过 docs ci 检查，没有需要检查的文档文件

ascend-robot成员

6月29日删除了label：docs-ci-pipeline-running

ascend-robot成员

6月29日添加了label：docs-ci-pipeline-success

ascend-robot成员

6月29日删除了label：ci-pipeline-running

ascend-robot成员

6月29日添加了label：ci-pipeline-passed

ascend-robot成员

6月29日评论：

流水线 PR-pipeline_pytorch#39888 [ commitID：28adc08a ] 已完成

>>>代码风格自动修复执行成功（无修复内容）

阶段	任务名	状态	详情
编译构建	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	🛑	>>>
	UT_ARM_A2_Select_Part_01	✅	>>>
	UT_ARM_A2_Select_Part_02	✅	>>>
流水线	PR-pipeline_pytorch	✅	>>>

此流水线已支持下列评论快捷指令，仅PR创建者和白名单成员[wujinyuan1, huangjingwei, liangsongwei, yashi999, culechan, Dring, wuyouqi1, L1919_snow, qq_52711437, WhiteNight12, nomiz, xiu_21, ffmh, wanglijun55, hss-shuai, husichao, smallsilly, lanshaozuishuai, jimmyisme1, lzy0920232, alpha-junh, Sunshine_Youngster, wei_zhuoyi, zhangyihuiben, zyw-hw, zzzkeke, rmch, yangch0324, LucciC, AACAES, renyujin, wjlflyer, senzhen-town, pengjingyou, qsc97, limuan, yule100, xiaoqi-zhou, kuhn7, chenxingying, hanye02, zichun_ye, anyrenwei, kkjocker, wangzili121, Lu_G, yvjc, puddingfjz, HandsoemLemon, bigprestigee1, huawuyi, zhenyu10, dairenjie, du-jin-hang, zou-jieyu, adelaideliu, TrHan, wanlinan, Windwindzzz, pengqihw, kisnwang, yuheng_wang, honghao_wang, jizewei, zhangguoguang, sunyu-xuan, chenrayray, hbhu_bin, liujunzhu, c_34, LiNuoh, maoyuanpeng1, zzhongmin, zhaoyu65, bellatan, jiabaolin, zhuofanshen, wencaiwen, lu_zhuge, caoshuyang, molly12, lyx324521, LQ1206, gitcode-chenjiao, cai-weiwei1989, CHDong, ogqin, yuanlipingGit, xuqinglin1, lqz2, zouwei1, chaoluoa, paradox325, jackzhang1116, yaoyao, akh, yujiacheng, dengjie0116, Hubert11111, Shine_Ws, wslhj555, longqiand, OYtao666, JiaqingQiang, luyyyy, Kingbelial, zhanghaiyu0101, wenxp1018, yanliu-luoluo, ksun_sekiro, liyong328, wgzheng, tangky, vivi_is_coding, aoiaoisola, weixin_44494597, wangmengmengwang65667, hid57809721, qq_35468730, comeonup, C547032, gcw_m5OQChA4, yao_yao_ling_xian, cnnbwcy, szqfes_12, cora_19, cann_lilin, can]评论有效

compile、compile_inductor、compile_torchair : 运行流水线
retry : 重试流水线所有失败子任务
retry <任务名> : 仅重试指定失败子任务
stop : 停止流水线

ascend-robot成员

6月29日评论：

流水线 PR-pipeline_pytorch#39888 [ commitID：28adc08a ] 已完成

>>>代码风格自动修复执行成功（无修复内容）

阶段	任务名	状态	详情
编译构建	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	🛑	>>>
	UT_ARM_A2_Select_Part_01	✅	>>>
	UT_ARM_A2_Select_Part_02	✅	>>>
流水线	PR-pipeline_pytorch	✅	>>>

compile、compile_inductor、compile_torchair : 运行流水线
retry : 重试流水线所有失败子任务
retry <任务名> : 仅重试指定失败子任务
stop : 停止流水线

Jingwei Huang成员

6月29日评论：

/lgtm

suhaibo成员

6月29日评论：

/approve

ascend-robot成员

6月29日添加了label：approvedlgtm

ascend-robot成员

6月29日合入了pull request

ascend-robot成员

6月29日评论：

流水线 pytorch_gitcode_PR_multiVersion#11544 [ commitID：28adc08a ] 已完成