算子GPU性能优化

[zhangting2020]

问题描述 Please describe your issue

问题描述

本任务为飞桨框架优化一系列算子的GPU计算性能，以提升模型训练效率。任务要求如下：

• 理解并分析现有算子的计算流程和性能瓶颈
• 设计优化方案并实现
• 使用op-benchmark工具进行算子性能的测试评估，比较优化前后的算子性能

注意事项

• 提交流程：请参考算子性能优化&提交流程，开发请参考
• RFC要求：可以不提交。但仍然建议参考链接中的算子优化文档去设计方案，在开发完成后自愿选择提交。
• 使用op-benchmark测试算子前、反向算子性能，前、反向算子的优化PR可以分开提交
• 认领规则：直接在 issue 下回复认领的任务 ID
• 提交PR前请参照官网安装pre-commit，检查代码格式。否则CI可能无法通过。
• PR 通过 CI 后，可以评论里或者 review request @zhangting2020 ，研发会进行审核
• 欢迎联系花花加入社区，和我们一起快乐开源！

任务序号	算子	认领人	提交PR
1	argmax和argmin	@AndSonder
2	argsort	@GreatV
3	cumsum	@enkilee
4	interp_trilinear	@GreatV
5	max_pool2d	@sanbuphy
6	normalize	@Difers
7	pad2d	@TanWei
8	pad3d	@jinyouzhi
9	affine_grid	@yangguohao
10	avg_pool2d	@chenhao-stick-to

[sanbuphy]

认领max_pool2d

[AndSonder]

序号	算子名称	认领人	PR
1	argmax和argmin	@AndSonder

[GreatV]

任务序号	算子	认领人	提交PR
2	argsort	@GreatV
4	interp_trilinear	@GreatV

[Difers]

序号	算子名称	认领人	PR
6	normalize	@Difers

[longranger2]

@enkilee 认领任务3

[TanWei]

认领任务7: pad2d

[luotao1]

@AndSonder 任务1可参考：

• 【PaddlePaddle Hackathon 3 No.37】为 Paddle 优化 argmin_argmax op 在 GPU 上的计算性能  #46655

[chenhao-stick-to]

认领任务10:avg_pool2d

[yangguohao]

认领任务9：affine_grid

[jinyouzhi]

认领8

[paddle-bot]

Since you haven't replied for more than a year, we have closed this issue/pr.
If the problem is not solved or there is a follow-up one, please reopen it at any time and we will continue to follow up.
由于您超过一年未回复，我们将关闭这个issue/pr。
若问题未解决或有后续问题，请随时重新打开，我们会继续跟进。