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

[Arm] rewrite reduce op for ARM #10148

Merged
merged 9 commits into from
Apr 11, 2023
Merged

[Arm] rewrite reduce op for ARM #10148

merged 9 commits into from
Apr 11, 2023

Conversation

mjp9527
Copy link
Collaborator

@mjp9527 mjp9527 commented Mar 28, 2023

PR devices

Arm

PR types

Bug fixes

PR changes

Backends

Description

重新整理 reduce 端代码,补齐缺少的后端,目前支持输入最多可达6维

@paddle-bot
Copy link

paddle-bot bot commented Mar 28, 2023

Thanks for your contribution!

@mjp9527 mjp9527 changed the title [Arm] fix reduce_n bug for memory leak [Arm] rewrite reduce op for ARM Apr 4, 2023
@@ -12,14 +12,32 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License. */

#include "lite/backends/arm/math/reduce_min.h"
#pragma once
Copy link
Collaborator

Choose a reason for hiding this comment

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

文件名改成 reduce.h 或 reduce_ops.h 吧

Copy link
Collaborator

@hong19860320 hong19860320 left a comment

Choose a reason for hiding this comment

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

LGTM

@mjp9527 mjp9527 merged commit ec1eb86 into PaddlePaddle:develop Apr 11, 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.

2 participants