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

【SCU】[Paddle TensorRT No.63] Add pd_op.mish converter #69705

Merged
merged 15 commits into from
Dec 4, 2024

Conversation

mori0umi
Copy link
Contributor

PR Category

Inference

PR Types

New features

Description

新增了pd_op.mishMarker和Converter

Copy link

paddle-bot bot commented Nov 26, 2024

你的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.

@luotao1 luotao1 added the HappyOpenSource Pro 进阶版快乐开源活动,更具挑战性的任务 label Nov 26, 2024
@@ -143,5 +143,19 @@ def test_trt_result(self):
self.check_trt_result()


class TestMishFloatTRTPattern(TensorRTBaseTest):
Copy link
Contributor

Choose a reason for hiding this comment

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

再加几个单测,测试下1维度,4维度和3维度能不能通过,这个是因为旧ir下是通过通用plugin的方式进入trt的,我们还未支持,如果你这个测对了,可以合入

@@ -158,5 +158,61 @@ def test_trt_result(self):
self.check_trt_result()


class TestMishCase1TRTPattern(TensorRTBaseTest):
Copy link
Contributor

Choose a reason for hiding this comment

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

输入支持int64吗,如果支持也需要加一下

Copy link
Contributor Author

Choose a reason for hiding this comment

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

查阅了paddle文档,仅支持float32和float64

@luotao1 luotao1 changed the title [Paddle TensorRT No.63] Add pd_op.mish converter 【SCU】[Paddle TensorRT No.63] Add pd_op.mish converter Dec 4, 2024
@luotao1 luotao1 merged commit 74f83fd into PaddlePaddle:develop Dec 4, 2024
28 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
contributor External developers HappyOpenSource Pro 进阶版快乐开源活动,更具挑战性的任务
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants