已合并
TORCH MAIN SYNC : unified kernel macro (AMP) && checkpoint error message improvement #11520
dilililiwhy创建于 2024年5月6日
TORCH MAIN SYNC : unified kernel macro (AMP) && checkpoint error message improvement #11520
已合并
dilililiwhy创建于 2024年5月6日
refs/pull/11520/head合入到master
dilililiwhy
dilililiwhy成员
2024年5月6日

What type of PR is this?

Uncomment only one /kind <> line, hit enter to put that in a new line, and remove leading whitespaces from that line:

/kind bug
/kind task
/kind feature

What does this PR do / why do we need it:

  • KERNEL_PRIVATEUSEONE
  • set_autocast_dtype/get_autocast_dtype
  • set_autocast_enabled/is_autocast_enabled

Which issue(s) this PR fixes:

Fixes #

Special notes for your reviewers:
部分宏及接口归一(AMP)
https://github.com/pytorch/pytorch/pull/124050
https://github.com/pytorch/pytorch/pull/124359
梯度检查点异常信息刷新
https://github.com/pytorch/pytorch/pull/125155

likedislike
Pull Request已成功合入, 合并人@
(感谢 dilililiwhy 的贡献)
dilililiwhy
dilililiwhy成员
2024年5月6日 评论:

compile

likedislike
atlasaccount1
atlasaccount1
2024年5月6日 评论:

流水线任务触发成功,正在执行,请稍候。 任务链接[a82e2d4d3d984c32adfd5b270cf4ddbe]

likedislike
atlasaccount1
atlasaccount1
2024年5月6日 评论:


SCA扫描: success

# check type result report
1 sca success >>>

likedislike
dilililiwhydilililiwhy成员
2024年5月6日 更新了说明
atlasaccount1
atlasaccount1
2024年5月6日 评论:


静态检查: pass

# check type result report
1 codecheck pass >>>

likedislike
此处折叠了62条消息 查看更多
atlasaccount1
atlasaccount1
2024年5月7日 评论:


SCA扫描: success

# check type result report
1 sca success >>>

likedislike
atlasaccount1
atlasaccount1
2024年5月7日 评论:


静态检查: pass

# check type result report
1 codecheck pass >>>

likedislike
misty-rain-typhoid
2024年5月8日 评论:

/lgtm

likedislike
huangyunlong成员
2024年5月8日 评论:

/lgtm

likedislike
du-jin-hang
du-jin-hang成员
2024年5月8日 评论:

/lgtm

likedislike