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

move REGISTER_OP_CUDA_KERNEL into cpp with eigen #32114

Conversation

Avin0323
Copy link
Contributor

@Avin0323 Avin0323 commented Apr 7, 2021

PR types

Others

PR changes

Others

Describe

#31851 优化编译修改的OP,对应的CUDA Kernel注册移动到cc文件中进行。

编译时间变化:

*.o 编译时间变化 变化比例
expand_as_op.cc.o 23s -> 27s 增加4s,增加17%
expand_as_op.cu.o 40s -> 0s 减少40s,减少100%
expand_as_v2_op.cc.o 25s -> 31s 增加6s,增加24%
expand_as_v2_op.cu.o 42s -> 0s 减少42s,减少100%
expand_op.cc.o 26s -> 32s 增加6s,增加23%
expand_op.cu.o 48s -> 0s 减少48s,减少100%
expand_v2_op.cc.o 30s -> 40s 增加10s,增加33%
expand_v2_op.cu.o 54s -> 0s 减少54s,减少100%
meshgrid_op.cc.o 23s -> 28s 增加5s,增加22%
meshgrid_op.cu.o 41s -> 0s 减少41s,减少100%
tile_op.cc.o 28s -> 35s 增加7s,增加25%
tile_op.cu.o 50s -> 0s 减少50s,减少100%

@paddle-bot-old
Copy link

paddle-bot-old bot commented Apr 7, 2021

Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

Copy link
Contributor

@wanghuancoder wanghuancoder left a comment

Choose a reason for hiding this comment

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

LGTM

@Avin0323 Avin0323 changed the title [WIP]move REGISTER_OP_CUDA_KERNEL into cpp with eigen move REGISTER_OP_CUDA_KERNEL into cpp with eigen Apr 20, 2021
@wanghuancoder wanghuancoder merged commit f6f59e5 into PaddlePaddle:develop Apr 20, 2021
@Avin0323 Avin0323 deleted the move-REGISTER_OP_CUDA_KERNEL-into-cpp-with-eigen branch April 20, 2021 06:57
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