Skip to content

Support fp16 for index_select and index_add#45601

Merged
limin2021 merged 37 commits intoPaddlePaddle:developfrom
limin2021:support_fp16_index_select_and_add
Sep 14, 2022
Merged

Support fp16 for index_select and index_add#45601
limin2021 merged 37 commits intoPaddlePaddle:developfrom
limin2021:support_fp16_index_select_and_add

Conversation

@limin2021
Copy link
Copy Markdown
Contributor

@limin2021 limin2021 commented Aug 31, 2022

PR types

Others

PR changes

OPs

Describe

(1) check of inputs on python side especially on dygraph mode will introduce some overhead, so we move them to to c++ backend.
(2) add fp16 support for index_add kernel.

@paddle-bot
Copy link
Copy Markdown

paddle-bot Bot commented Aug 31, 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.

@limin2021 limin2021 merged commit 61012a7 into PaddlePaddle:develop Sep 14, 2022
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.

2 participants