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 OP&Test] add bf16 fp16 type support for expand_v2_op and top_k_v2_op #51263

Merged
merged 14 commits into from
Apr 11, 2023
Merged

[AMP OP&Test] add bf16 fp16 type support for expand_v2_op and top_k_v2_op #51263

merged 14 commits into from
Apr 11, 2023
+151 −10

Conversation

HexToString
Copy link
Contributor

PR types

Others

PR changes

OPs

Describe

add bf16 fp16 type support for expand_v2_op and top_k_v2_op

@paddle-bot
Copy link

paddle-bot bot commented Mar 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.

ZzSean
ZzSean reviewed Apr 10, 2023
View reviewed changes
python/paddle/fluid/tests/unittests/test_expand_v2_op.py
self.op_type = "expand_v2"
self.prim_op_type = "prim"
self.dtype = np.float16
self.python_api = paddle.expand
Copy link
Contributor

Choose a reason for hiding this comment

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

根据CI报错，需要加一下public_python_api
AssertionError: If you want to check prim, please set public_python_api in setUp function.

ZzSean
ZzSean approved these changes Apr 11, 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

@ZzSean ZzSean merged commit 5b09dd5 into PaddlePaddle:develop Apr 11, 2023
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

@luotao1 luotao1 luotao1 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.
3 participants
@HexToString @luotao1 @ZzSean