已合并
[feat] fusedsgd #5277
culechan创建于 6月26日
[feat] fusedsgd #5277
已合并
culechan创建于 6月26日
culechan
culechan成员
6月26日

【合入来源】

如有社区issue,请关联issue链接
请勿携带内部流程信息(需求链接、问题单、内部issue等)

【修改方案】

增加对aten::fused_sgd_以及_fused_sgd.tensor_lr的适配;相比GPU的方案,由于NPU架构的差异性,把对found_inf校验和取lr值放在了host侧做。

【资料变更】

不涉及

【接口变更】

不涉及

【功能验证】

本地验证通过,由于cann包未上门禁,先skip
image.png

【CheckList】

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

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 culechan 的贡献)
culechanculechan成员
6月26日 创建了 pull request,commit e9642626
ascend-robotascend-robot成员
6月26日 添加了label:ascend-cla/yes
ascend-robot
ascend-robot成员
6月26日 评论:

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
repo-Ascend/op-plugin liangsongwei, daifu1234567 (2/2) liangsongwei (1/1)

💡 Tip:

  • Committer can comment /approve or /lgtm
  • Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

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

likedislike
ascend-robotascend-robot成员
6月26日 添加了label:needs-issue
此处折叠了246条消息 查看更多
culechan
culechan成员
7月1日 评论:

/check-issue

likedislike
ascend-robotascend-robot成员
7月1日 删除了label:needs-issue
ascend-robotascend-robot成员
7月1日 关闭了关联的issue
ascend-robotascend-robot成员
7月1日 合入了pull request
ascend-robot
ascend-robot成员
7月1日 评论:

Pull Request 已合并或已关闭。

If you want to solve this problem, you can click here to do it in the FAQs.

likedislike