已开启
【API一致性任务】8月社区任务第三期-Ascend for PyTorch API 一致性开发(71) #211
群青世界创建于  8月20日
群青世界成员
8月20日 创建

一、任务概览
Ascend for PyTorch是昇腾 NPU 适配 PyTorch 框架的核心插件,为 PyTorch 提供完整的昇腾硬件算子支持、调度能力与推理优化,保障 PyTorch 模型可在昇腾 NPU 上无缝、高效运行。
当前 PyTorch 官方原生 API 已形成完善体系,但部分API尚未在 Ascend for PyTorch 中完成实现、对齐与验证,导致部分 PyTorch 原生模型 / 代码在昇腾 NPU 上运行时出现接口报错、功能不一致等问题。
本任务邀请开发者参与Ascend for PyTorch API 补齐与适配验证,完成缺失 API 的开发、调试、测试与文档交付,完善 Torch-NPU 对 PyTorch 原生 API 的兼容性,提升昇腾 NPU 上 PyTorch 框架的易用性与模型适配覆盖率。
二、任务要求及参考指导
https://docs.qq.com/doc/DT3R3eVNWdm1yUkdR
三、任务价值
深度参与昇腾生态 + PyTorch 开源共建,贡献代码合入官方仓库;
掌握 NPU PyTorch 框架适配、异构计算核心技能;
获得开源社区认可,积累工业级实习项目经验,提升简历竞争力。
四、任务领取
请前往昇腾社区-任务广场领取任务:https://www.hiascend.com/activities/task-center ,其余领取方式无效。(昇腾社区任务发布会滞后,请耐心等候)
五、API名称
torch.autograd.grad_mode.set_multithreading_enabled.clone

likedislike
TorchNPU-BotTorchNPU-Bot
8月21日 添加了label:help-wantedevent: api-consistencybot-triaged
TorchNPU-Bot
TorchNPU-Bot
8月21日 评论:

检测到社区任务相关 issue,自动添加标签:event: api-consistency、help-wanted

likedislike
shashasha123
9月21日 评论:

API 一致性任务:set_multithreading_enabled.clone 支持情况验证

本草稿汇总 A2、A3、A5 环境中的聚焦验证结果。

验证范围

目标 API 是 torch.autograd.grad_mode.set_multithreading_enabled.clone。
PyTorch 2.7.1 官方用例为
test/test_autograd.py::TestAutograd::test_set_multithreading_enabled_as_context_manager_and_function。

聚焦检查覆盖以下行为:

  • False 和 True 模式返回同类 clone 对象。
  • clone 保留 mode。
  • 上下文退出后恢复进入上下文前的多线程状态。

A2 结果

  • Python:3.11.15。
  • PyTorch:2.10.0+cpu。
  • Torch-NPU:2.10.0。
  • NPU:1 张 910B3,可用。
  • 结果:聚焦断言通过,退出状态恢复正确。
  • 证据:/workspace/aiframework/results/a2-api-71.txt。
  • 截图:
    72e2594e3eed4ac2b7e52c63a0c0fbb8.png

A5 结果

  • Python:3.11.15。
  • PyTorch:2.7.1+cpu。
  • Torch-NPU:2.7.1.post8。
  • NPU:1 张 Ascend950PR,可用。
  • 结果:同一聚焦断言通过,exit code 为 0。
  • 证据:private/remote-results/a5/a5-api71-run.log。
  • 官方测试尝试因精简源码缺少 autograd 夹具未完成。
  • A5 截图:
    0505bf8aaaa54da19200d1bd1ab84f8d.png

A3 结果

  • Python:3.11.15。
  • PyTorch:2.10.0+cpu。
  • Torch-NPU:2.10.0。
  • NPU:2 张 Ascend910,可用。
  • 结果:同一聚焦断言通过,exit code 为 0。
  • 证据:private/remote-results/a3/a3-api71-run.log。
  • 官方测试因精简源码缺少 test/test_autograd.py 未启动。
  • A3 截图:
    cd2015e6784541bbb9c7d8f9abcce119.png
    官方测试文件在当前精简上传包中缺少 test/autograd 夹具,无法完成文件级
    启动;上述等价断言已单独执行并通过。

NPU 适配结论

上游实现位于 torch/autograd/grad_mode.py,目标 API 只管理自动求导的 Python
上下文状态和 clone 对象,不创建 NPU 张量,也不依赖设备特有算子。因此无需 NPU
专用 patch;本次用 A2、A3、A5 运行聚焦断言确认行为一致。

native API 文档支持情况

已检查 master 分支的 native API 文档目录。目标 API 已有对应条目:

因此本任务不需要另提文档 PR。

结论

当前 A2、A3、A5 证据支持该纯 Python API 的行为一致性;官方文件级测试夹具缺失情况
已记录,等价聚焦断言在三套环境均通过。

likedislike
shashasha123
9月28日 评论:

@TorchNPU-Bot @Giotto0025 您好,我的验收材料已经提交多日了,能否处理一下,不然就快到截止时间了

likedislike
shashasha123
9月29日 评论:

/compile

likedislike
xyzllm2
xyzllm2
23 天前 评论:

@shashasha123 此任务的到期时间为2026年10月21日,还请及时提交并在微信群中催促审批,如没有加入群聊,或有相关事宜需要咨询可以通过邮箱或者微信联系我;邮箱:1624640779@qq.com;微信:sh-xyzllm-7864 也可以通过手机号:17570017864 加我微信

likedislike