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.

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

[AMP OP&Test] Support bf16/fp16 for dropout and full_like op #51809

Merged
merged 5 commits into from
Mar 29, 2023

Conversation

westfish
Copy link
Contributor

PR types

New features

PR changes

Others

Describe

Support bf16/fp16 for dropout and full_like op.

@westfish westfish changed the title [AMP OP&Test] Support bf16/fp16 for dropout and full_like op. [AMP OP&Test] Support bf16/fp16 for dropout and full_like op Mar 18, 2023
@westfish westfish requested a review from ZzSean March 21, 2023 07:42
@westfish
Copy link
Contributor Author

@Vvsmile @ZzSean 请问还有什么要修改的地方吗? 关于dropout的API Check Dtype这一项任务我检查过了,已经存在check_variable_and_dtype的调用和检查FP16类型了,就没有进行修改:

check_variable_and_dtype(

check_variable_and_dtype(x, 'x', ['float32', 'float64'], 'dropout')

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

@westfish westfish requested a review from luotao1 March 29, 2023 03:08
@ZzSean ZzSean merged commit 1221893 into PaddlePaddle:develop Mar 29, 2023
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.

4 participants