已关闭
【社区任务】Ascend for PyTorch测试用例解耦(20)【已领取】 #20
ltllt1创建于  8月5日关闭于  22 天前
ltllt1成员
8月5日 创建

一、任务概览
PyTorch上游社区大量单元测试用例硬编码绑定CUDA设备,存在设备写死、CUDA专属装饰器、CUDA专属API、 pytest硬编码is_cuda参数、仅CPU默认创建Tensor等问题。导致昇腾NPU等异构加速器无法复用社区原生测试用例、无法接入社区设备自动化测试框架,存在硬件绑定耦合高、可移植性差、新硬件适配成本高的问题。
本任务核心目标是邀请Ascend for PyTorch社区开发者参与PyTorch上游社区测试用例硬件解耦适配任务,在不改变社区原有测试语义、不增减社区用例数量、不破坏原有执行逻辑的前提下,完成存量测试用例硬件解耦改造,实现测试用例设备无关化,支持昇腾 NPU动态调度运行,对齐社区官方设备适配规范。

二、任务要求及参考指导
1、开发者指导书
https://docs.qq.com/doc/DUXpVSHNERFJJaXBH

2、PyTorch社区官方测试用例解耦跟踪表:
https://docs.google.com/spreadsheets/d/1cDNiLW4KvPcGYPlA3KCDm0zV5PLPUWubno1OyCznKBw/edit?pli=1&gid=1919673129#gid=1919673129

重要说明:开发者成功领取任务后,请第一时间在此表格中“Assignee”列中登记本人GitHub ID(注意不要覆盖已有id)。由于本次任务是PyTorch社区面向所有开发者公开招募的,因此可能存在同一个测试用例文件被多人领取的情况。所有任务采用先合入优先归属规则,同一测试用例以率先成功合入PyTorch社区上游PR的开发者为准,最终成果归属于该开发者。请开发者领取任务后务必尽快完成开发和评审,需要在其他社区开发者之前完成合入,昇腾社区任务才能验收通过。

三、任务领取
请前往昇腾社区-任务广场领取任务:https://www.hiascend.com/activities/task-center ,其余领取方式无效。(昇腾社区任务发布会滞后,请耐心等待)

四、任务内容
test/test_pruning_op.py

likedislike
Hhid87721478成员
8月13日 添加了label:help-wanted
Hhid87721478成员
8月13日 添加了label:triaged
Jinfan Liu
Jinfan Liu
8月18日 评论:

【test/test_pruning_op.py 解耦 PR 登记(更新)】

上游 PR: https://github.com/pytorch/pytorch/pull/194193
GitHub ID: JfanLiu
PR 状态: Draft;标签: module: tests/others

场景判定: 场景三(CPU only,无需 device 解耦;按指导会纪要加语义标签 hw_classification = HardwareClassification.GENERIC)

改造内容: test/test_pruning_op.py 新增 HardwareClassification 导入,PruningOpTest 类标注 hw_classification = HardwareClassification.GENERIC(+7/-1;不改语义、不增减用例、不改执行逻辑)

双环境验证结论(均已通过):

  • 910B/910C 侧(910B3): 基线 Ran 2 tests OK;改造后 Ran 2 tests OK(CPU 与 NPU 均通过)
  • 950 侧(2026-08-18 补跑): 基线 Ran 2 tests OK;GENERIC 过滤 2/2 命中;ACCELERATOR 过滤 0/2 排除;pytest 2 passed / no tests ran
likedislike
Hhid87721478成员
8月26日 修改标题为 “【社区任务】Ascend for PyTorch测试用例解耦(20)【已领取】”,原标题为“【社区任务】Ascend for PyTorch测试用例解耦(20)”
Jinfan LiuJinfan Liu
9月15日 关联了pull request:【社区任务】Ascend for PyTorch测试用例解耦(20) 交付件
ascend-robotascend-robot成员
22 天前 关闭了 issue
ascend-robotascend-robot成员
22 天前 issue状态由 TODO 改变为 CLOSED
ascend-robotascend-robot成员
22 天前 添加了label:resolved