已合并
test(fx): add Interpreter internal API test cases for NPU [v2.7.1] #43909
zkx创建于 14 天前
test(fx): add Interpreter internal API test cases for NPU [v2.7.1] #43909
已合并
Pull Request已成功合入, 合并人@ascend-robot
(感谢 zkx 的贡献)atomgit-bot
14 天前 评论:
14 天前 评论:
变更摘要
此 PR 新增了 test/test_fx_interpreter_npu.py 测试文件,针对 torch.fx.Interpreter 的四个内部方法(boxed_run、fetch_attr、map_nodes_to_values、fetch_args_kwargs_from_env)补充 NPU 环境下的直接验证用例。PyTorch 官方社区对这些方法缺少专门测试,本文件填补了该空白。所有用例通过 torch.accelerator.current_accelerator() 自适应获取当前加速器类型,NPU 不可用时自动回退至 CPU,确保张量在对应设备上执行。
主要改动
- 新增
TestInterpreterBoxedRun测试类:包含test_boxed_run_basic和test_boxed_run_clears_args两个用例,分别验证boxed_run执行结果正确性以及参数列表在使用后被清空的边界行为。 - 新增
TestInterpreterFetchAttr测试类:包含test_fetch_attr_parameter和test_fetch_attr_submodule,验证fetch_attr按属性路径获取Parameter和子模块的功能。 - 新增
TestInterpreterMapNodesToValues测试类:包含test_map_nodes_to_values_args和test_map_nodes_to_values_kwargs,验证map_nodes_to_values将图节点映射为具体张量值的能力,覆盖位置参数元组和关键字参数字典两种返回形式。 - 新增
TestInterpreterFetchArgsKwargsFromEnv测试类:包含test_fetch_args_kwargs_from_env,验证从执行环境env中提取节点的位置参数与关键字参数的完整流程。


ascend-robot
14 天前 评论:
14 天前 评论:
atomgit-bot
14 天前 评论:
14 天前 评论:
此处折叠了90条消息 查看更多
梁松伟
5 天前 评论:
5 天前 评论:
/approve


5 天前 添加了label:approvedlgtm
5 天前 删除了label:ci-pipeline-passed
5 天前 合入了pull request
ascend-robot
4 天前 评论:
4 天前 评论:
流水线 pytorch_gitcode_PR_multiVersion#14069 [ commitID:2f4fcbd1 ] 已完成


【合入来源】
【修改方案】
一、API功能
二、测试用例完备性说明
三、NPU适配
3.1 API适配:无需修改。4 个方法均为纯框架层图解释逻辑,与底层硬件解耦。
3.2 测试用例适配:新增用例在 import 后添加设备获取命令 device_type = acc.type if (acc := torch.accelerator.current_accelerator()) else "cpu",
全部张量通过 .to(device_type) 运行在 NPU 上(CPU 环境回落 cpu)。
【资料变更】
【接口变更】
【功能验证】
【CheckList】