-
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
add nearest_interp_v2 trt plugin #34126
add nearest_interp_v2 trt plugin #34126
Conversation
Thanks for your contribution! |
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.
please add UT
bcd60f7
to
5d617a8
Compare
e1802de
to
c8e7d02
Compare
paddle/fluid/inference/tensorrt/convert/nearest_interp_v2_op.cc
Outdated
Show resolved
Hide resolved
python/paddle/fluid/tests/unittests/ir/inference/test_trt_nearest_interp_v2_op.py
Outdated
Show resolved
Hide resolved
b1128e3
to
90b8a52
Compare
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.
LGTM for infer_ut
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.
LGTM for set_tests_properties(test_trt_nearest_interp_v2_op PROPERTIES TIMEOUT 30)
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.
LGTM
PR types
New features
PR changes
Others
Describe
add nearest interp v2 trt plugin converter, the inference speed of ppyolov2 r50vd dcn before and after this PR can be seen as follows: