已合并
fix(nn): fix test for nn api: torch.nn.ParameterDict, torch.nn.ParameterList, torch.nn.Sequential #32056
dinglaiping创建于 3月19日
fix(nn): fix test for nn api: torch.nn.ParameterDict, torch.nn.ParameterList, torch.nn.Sequential #32056
已合并
dinglaiping创建于 3月19日
已删除 :fixtest-nn-api-2.7.1合入到Ascend/pytorchv2.7.1
dinglaiping成员
3月19日

【合入来源】

https://gitcode.com/Ascend/pytorch/issues/1585
请勿携带内部流程信息(需求链接、问题单、内部issue等)

  • issue/工单

【修改方案】

请描述修改内容的具体实现,涉及哪些组件之间进行交互,可以用1、2、3、...进行罗列
如果是需求或者重构类的PR,需要补充详细设计文档(说明上下游组件关系、时序图、类图、DFX能力等内容)

  1. 2.6.0版本提示skipIfMps找不到,要换成skipIfMPS。
  2. 2.7及以上版本skipIfMps找不到,换成skipIfMPS;new_module_tests找不到,需要把导入new_module_tests改成导入get_new_module_tests并在调用的地方换成get_new_module_tests();tf32_on_and_off找不到,删掉导入,并在调用tf32_is_not_fp32()的地方换成调用torch.cuda.is_tf32_supported()

【资料变更】

不涉及

【接口变更】

不涉及

【功能验证】

说明测试场景,测试方法。如果本次测试方式与常规单元测试不同,请详细说明您的测试步骤
新增/变更内容是否已新增/适配UT测试用例看护,并补充测试自验证截图

整体结论:
本次验证通过修复官网test_nn.py用例文件,验证3个API,功能正常:
ParameterDict:官方用例中键值对索引、新增、删除、遍历等操作均正常,容器内所有参数设备统一为npu:0,无设备漂移问题;
ParameterList:官方用例中数字索引、append新增、遍历等操作均正常,参数设备一致性与长度校验均通过,有序管理能力正常;
Sequential:官方用例中getitem/setitem/delitem/append/insert/extend/pop等所有核心操作均正常执行,操作后子模块设备均保持为npu:0,动态扩展与元素管理能力完整。

运行日志:
[root@hostname-fqv42 torchnpuapi]# docker exec -it test27 bash
root@hostname-fqv42:/home# cd /root/torchnpuapi
root@hostname-fqv42:~/torchnpuapi# python -m unittest test_nn_npu27.TestNN.test_ParameterList test_nn_npu27.TestNN.test_ParameterDict test_nn_npu27.TestNN.test_Sequential_getitem test_nn_npu27.TestNN.test_Sequential_setitem test_nn_npu27.TestNN.test_Sequential_delitem test_nn_npu27.TestNN.test_Sequential_append test_nn_npu27.TestNN.test_Sequential_insert test_nn_npu27.TestNN.test_Sequential_extend test_nn_npu27.TestNN.test_Sequential_pop -v
test_ParameterList (test_nn_npu27.TestNN.test_ParameterList) ... ok
test_ParameterDict (test_nn_npu27.TestNN.test_ParameterDict) ... ok
test_Sequential_getitem (test_nn_npu27.TestNN.test_Sequential_getitem) ... ok
test_Sequential_setitem (test_nn_npu27.TestNN.test_Sequential_setitem) ... ok
test_Sequential_delitem (test_nn_npu27.TestNN.test_Sequential_delitem) ... ok
test_Sequential_append (test_nn_npu27.TestNN.test_Sequential_append) ... ok
test_Sequential_insert (test_nn_npu27.TestNN.test_Sequential_insert) ... ok
test_Sequential_extend (test_nn_npu27.TestNN.test_Sequential_extend) ... ok
test_Sequential_pop (test_nn_npu27.TestNN.test_Sequential_pop) ... ok


Ran 9 tests in 0.057s

OK
root@hostname-fqv42:~/torchnpuapi#

【CheckList】

PR提交人对以下CheckList自检项进行全量自检,自检通过或不涉及,均修改 [ ] 为 [x]

  • 代码注释完备,正确记录错误日志
  • 代码实现进行了返回值、空指针等校验
  • PR标题正确使用类型标签,如:feat、fix、refactor、docs、test等
  • PR持续集成流水线(CI)执行通过,代码检查无异常
likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 dinglaiping 的贡献)
Ddinglaiping成员
3月19日 创建了 pull request,commit 9f9640a2
Ddinglaiping成员
3月19日 关联了issue:[Bug]: 验证torch.nn.ParameterDict, torch.nn.ParameterList, torch.nn.Sequential这三个API的时候发现官网test目录下test_nn.py文件存在错误
dinglaiping成员
3月19日 评论:

compile

likedislike
ascend-robotascend-robot成员
3月19日 添加了label:ci-pipeline-running
此处折叠了161条消息 查看更多
haiyan8
haiyan8
3月25日 评论:

/lgtm

likedislike
liwei386成员
3月25日 评论:

/approve

likedislike
ascend-robotascend-robot成员
3月25日 添加了label:approvedlgtm
ascend-robot
ascend-robot成员
3月25日 评论:

Review Guide

This pull-request passes review.
Committers who wrote a comment of /approve are: liwei386.
Reviewers who wrote a comment of /lgtm are: liwei386, yuhaiyan.

likedislike
ascend-robotascend-robot成员
3月25日 合入了pull request