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

[Prim] Modify the reduce_as op #65002

Merged
merged 6 commits into from
Jun 11, 2024
Merged

Conversation

zeroRains
Copy link
Contributor

PR Category

Others

PR Types

Others

Description

现在reduce_as op的实现是在计算好reduce_dim之后将reduce_dim作为输入传给reduce_sum kernel。当reduce_dim为空数组时,reduce_sum会默认执行reduce_all的计算,但是在reduce_as op中,当reduce_dim为空时,期望是不对输入做任何操作。该PR就是修复reduce_as op在reduce_dim为空时的情况。

Copy link

paddle-bot bot commented Jun 9, 2024

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

@paddle-bot paddle-bot bot added the contributor External developers label Jun 9, 2024
@zeroRains zeroRains changed the title Modify the reduce_as op [Prim] Modify the reduce_as op Jun 9, 2024
@cyber-pioneer cyber-pioneer merged commit 2372288 into PaddlePaddle:develop Jun 11, 2024
32 of 33 checks passed
@zeroRains zeroRains deleted the reduce_as branch June 11, 2024 02:48
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
contributor External developers
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants