Skip to content

Conversation

zhoutianzi666
Copy link
Contributor

@zhoutianzi666 zhoutianzi666 commented Aug 9, 2022

PR types

Others

PR changes

Others

Describe

When filter is not a persistable tensor, forbid it entering into trt
对于slice,squeeze,unsqueeze,暂时禁止让int64_t的输入进入trt
对于scale,只让float和fp16进入trt

@paddle-bot
Copy link

paddle-bot bot commented Aug 9, 2022

你的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 contributor External developers status: proposed labels Aug 9, 2022
Copy link

@cyj1986 cyj1986 left a comment

Choose a reason for hiding this comment

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

LGTM

@zhoutianzi666 zhoutianzi666 changed the title [Paddle-TRT] fix conv2d [Paddle-TRT] fix conv2d/int64 Aug 10, 2022
Copy link

@cyj1986 cyj1986 left a comment

Choose a reason for hiding this comment

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

LGTM

@cyj1986 cyj1986 merged commit 9a04540 into PaddlePaddle:release/2.3 Aug 10, 2022
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