Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[AMP Prim OP]support some prim ops for bf16 dtype part5 #54422

Merged
merged 2 commits into from
Jun 8, 2023
Merged

[AMP Prim OP]support some prim ops for bf16 dtype part5 #54422

merged 2 commits into from
Jun 8, 2023

Conversation

Charles-hit
Copy link
Contributor

PR types

Others

PR changes

Others

Description

Pcard-66975
PR改动：

  • 支持elementwise_min、pad、roll、scatter、split、sequeeze2、unsequeeze2、statck、tile 组合算子 bfloat16 dtype并且添加相关单测。

@paddle-bot
Copy link

paddle-bot bot commented Jun 7, 2023

你的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.

ZzSean
ZzSean approved these changes Jun 8, 2023
View reviewed changes
Copy link
Contributor

@ZzSean ZzSean left a comment

Choose a reason for hiding this comment

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

LGTM for skipIf

@Charles-hit Charles-hit merged commit bb89c0c into PaddlePaddle:develop Jun 8, 2023
@Charles-hit Charles-hit deleted the support_bf16_for_primops_part5 branch June 8, 2023 03:33
Charles-hit added a commit to Charles-hit/Paddle that referenced this pull request Jun 13, 2023
@Charles-hit
[AMP Prim OP]support some prim ops for bf16 dtype part5 (PaddlePaddle…
41eb54f 
…#54422)

* support some prim ops for bf16 dtype

* remove useless code
@Charles-hit Charles-hit mentioned this pull request Jun 13, 2023
Merged
phlrain pushed a commit that referenced this pull request Jun 14, 2023
@Charles-hit
[cherry-pick]Fix cuda12 test (#54622)
bd1a240 
* [AMP Prim OP]support some prim ops for bf16 dtype part3 (#54368)

* support some prim ops bf16 dtype

* fix cmake

* [AMP Prim OP]support some prim ops for bf16 dtype part5 (#54422)

* support some prim ops for bf16 dtype

* remove useless code

* support some prim ops bf16 dtype (#54399)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@ZzSean ZzSean ZzSean approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@Charles-hit @ZzSean