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.

Sign up for GitHub

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

Jump to bottom

[Paddle-Inference] rebuild matmul pass: trt and gpu_cpu #39369

Merged
merged 4 commits into from
Feb 9, 2022
Merged

[Paddle-Inference] rebuild matmul pass: trt and gpu_cpu #39369

merged 4 commits into from
Feb 9, 2022

Conversation

Wangzheee
Copy link
Contributor

@Wangzheee Wangzheee commented Feb 7, 2022 •
edited
Loading

PR types

Others

PR changes

Others

Describe

1、将matmul融合相关的 pass 基于不同的硬件拆开，不同硬件（GPU、CPU、TensorRt）执行不同的pass；
2、在TensorRt的情况下增加 fc 权重转置的功能支持

@paddle-bot-old
Copy link

paddle-bot-old bot commented Feb 7, 2022

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

shangzhizhou
shangzhizhou approved these changes Feb 9, 2022
View reviewed changes
Copy link
Member

@shangzhizhou shangzhizhou left a comment

Choose a reason for hiding this comment

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

LGTM，请充分测试

heliqi
heliqi reviewed Feb 9, 2022
View reviewed changes
Copy link
Contributor

@heliqi heliqi left a comment

Choose a reason for hiding this comment

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

LGTM

@Wangzheee Wangzheee merged commit db7d129 into PaddlePaddle:develop Feb 9, 2022
@wozna wozna mentioned this pull request Feb 22, 2022
Closed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@heliqi heliqi heliqi left review comments

@raindrops2sea raindrops2sea raindrops2sea approved these changes

@shangzhizhou shangzhizhou shangzhizhou approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
4 participants
@Wangzheee @heliqi @raindrops2sea @shangzhizhou