Skip to content

[phi] Migrate amp XPU kernels to PHI#45569

Merged
YuanRisheng merged 2 commits intoPaddlePaddle:developfrom
jiahy0825:amp_xpu_kernel
Aug 31, 2022
Merged

[phi] Migrate amp XPU kernels to PHI#45569
YuanRisheng merged 2 commits intoPaddlePaddle:developfrom
jiahy0825:amp_xpu_kernel

Conversation

@jiahy0825
Copy link
Contributor

@jiahy0825 jiahy0825 commented Aug 30, 2022

PR types

Others

PR changes

Others

Describe

  • Migrate update_loss_scaling XPU kernel to PHI.
  • Migrate check_finite_and_unscale XPU kernel to PHI.
  • Note: update_loss_scaling and check_finite_and_unscale are both amp ops, migrate them through this PR.

@paddle-bot
Copy link

paddle-bot bot commented Aug 30, 2022

你的PR提交成功,感谢你对开源项目的贡献!
请关注后续CI自动化测试结果,详情请参考Paddle-CI手册
Your PR has been submitted. Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

Copy link
Contributor

@ykkk2333 ykkk2333 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm

@YuanRisheng YuanRisheng merged commit 68bfa0c into PaddlePaddle:develop Aug 31, 2022
@jiahy0825 jiahy0825 deleted the amp_xpu_kernel branch August 31, 2022 09:07
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants