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 part4 #54399

Merged
merged 1 commit into from
Jun 7, 2023
Merged

[AMP Prim OP]support some prim ops for bf16 dtype part4 #54399

merged 1 commit into from
Jun 7, 2023

Conversation

Charles-hit
Copy link
Contributor

@Charles-hit Charles-hit commented Jun 6, 2023 •
edited
Loading

PR types

Others

PR changes

Others

Description

Pcard-66975
PR改动：

  • 支持concat, gather_nd,elementwise_max,elementwise_pow,reduce_max组合算子bf16dtype并添加单测。
  • 修复gather_nd 单测bf16数值法步长设置过小导致单测挂的问题。
  • 支持sum kernel cpu bf16数据类型保障elementwise_max组合算子可以支持bf16类型。

@paddle-bot
Copy link

paddle-bot bot commented Jun 6, 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.

cyber-pioneer
cyber-pioneer approved these changes Jun 7, 2023
View reviewed changes
Copy link
Contributor

@cyber-pioneer cyber-pioneer left a comment

Choose a reason for hiding this comment

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

LGTM

@Charles-hit Charles-hit merged commit 791963a into PaddlePaddle:develop Jun 7, 2023
@Charles-hit Charles-hit deleted the support_bf16_for_primops_part4 branch June 7, 2023 07:06
Charles-hit added a commit to Charles-hit/Paddle that referenced this pull request Jun 13, 2023
@Charles-hit
support some prim ops bf16 dtype (PaddlePaddle#54399)
386fc7f
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

@cyber-pioneer cyber-pioneer cyber-pioneer 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 @cyber-pioneer