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
remove input dim check in op_teller and update ut #37097
remove input dim check in op_teller and update ut #37097
Conversation
Thanks for your contribution! |
@@ -176,12 +176,6 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8, | |||
<< " op does not support input's dim is 1 in tensorrt."; | |||
return false; | |||
} | |||
// TODO(inference): fix | |||
if (x_shape.size() == 2 && !with_dynamic_shape) { |
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.
针对二维输入的情况是如何修复的呢?pr已经合入了吗?
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.
应该是这个,#36614
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
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
Others
PR changes
Others
Describe
移除op_teller中一些op对2维输入的过滤,该问题已经被修复了,过滤会影响业务模型推理性能。
同时更新单测。