未验证 提交 be00a42f 编写于 作者: W weishengying 提交者: GitHub

Fix bugs in "test_custom_plugin_creater" unit test (#46075)

上级 d072aaeb
......@@ -103,13 +103,7 @@ nv_test(
nv_test(
test_custom_plugin_creater
SRCS test_custom_plugin_creater.cc
DEPS paddle_framework
${GLOB_OPERATOR_DEPS}
tensorrt_engine
tensorrt_plugin
tensorrt_converter
op_meta_info
custom_operator)
DEPS paddle_framework tensorrt_converter op_meta_info custom_operator)
if(WITH_ONNXRUNTIME AND WIN32)
# Copy onnxruntime for some c++ test in Windows, since the test will
......
......@@ -102,6 +102,8 @@ TEST(CustomPluginCreater, StaticShapePlugin) {
framework::Scope scope;
tensorrt::plugin::TrtPluginRegistry::Global()->RegistToTrt();
auto &custom_plugin_tell = OpTeller::Global().GetCustomPluginTeller();
framework::OpDesc custom_op(*op_desc, nullptr);
......@@ -186,6 +188,8 @@ TEST(CustomPluginCreater, DynamicShapePlugin) {
framework::Scope scope;
tensorrt::plugin::TrtPluginRegistry::Global()->RegistToTrt();
auto &custom_plugin_tell = OpTeller::Global().GetCustomPluginTeller();
framework::OpDesc custom_op(*op_desc, nullptr);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册