-
Notifications
You must be signed in to change notification settings - Fork 5.6k
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
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
@@ -143,5 +143,19 @@ def test_trt_result(self): | |||
self.check_trt_result() | |||
|
|||
|
|||
class TestMishFloatTRTPattern(TensorRTBaseTest): |
There was a problem hiding this comment.
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): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
输入支持int64吗,如果支持也需要加一下
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
查阅了paddle文档,仅支持float32和float64
PR Category
Inference
PR Types
New features
Description
新增了
pd_op.mish
Marker和Converter