-
Notifications
You must be signed in to change notification settings - Fork 5.8k
[Paddle TensorRT] add pd_op.max converter #67956
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
[Paddle TensorRT] add pd_op.max converter #67956
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
test/tensorrt/test_trt_marker_max.py
Outdated
@@ -0,0 +1,58 @@ | |||
# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved. |
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.
有了test_converter_max就不需要有这个文件了,删除
test/tensorrt/test_converter_max.py
Outdated
@@ -0,0 +1,41 @@ | |||
# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved. |
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.
CMakeLists.txt加一个限制时间的属性,否则就超时
test/tensorrt/test_converter_max.py
Outdated
@@ -0,0 +1,41 @@ | |||
# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved. | |||
# |
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.
该文件建议重命名为test_converter_math.py,后续所有math.py下的converter单测都会放到这里
test/tensorrt/CMakeLists.txt
Outdated
@@ -14,4 +14,5 @@ if(NOT WIN32 AND TENSORRT_FOUND) | |||
set_tests_properties(test_converter_conv2d PROPERTIES TIMEOUT "100") | |||
set_tests_properties(test_converter_add PROPERTIES TIMEOUT "100") | |||
set_tests_properties(test_converter_batch_norm PROPERTIES TIMEOUT "100") | |||
set_tests_properties(test_converter_max PROPERTIES TIMEOUT "100") |
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.
你不是叫test_converter_math吗
hi, @Layssy
|
PR Category
Inference
PR Types
New features
Description
card-71500
添加max op 的 marker 和 converter