File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed
paddle/fluid/inference/tensorrt/convert Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -18,7 +18,7 @@ nv_test(test_trt_activation_op SRCS test_activation_op.cc activation_op.cc
18
18
nv_test (test_trt_conv_op SRCS test_conv2d_op.cc conv2d_op.cc
19
19
DEPS ${FLUID_CORE_MODULES} ${GLOB_OPERATOR_DEPS} tensorrt_engine conv_op conv_transpose_op SERIAL )
20
20
nv_test (test_trt_pool2d_op SRCS test_pool2d_op.cc pool2d_op.cc
21
- DEPS ${FLUID_CORE_MODULES} ${GLOB_OPERATOR_DEPS} tensorrt_engine pool_op SERIAL )
21
+ DEPS ${FLUID_CORE_MODULES} ${GLOB_OPERATOR_DEPS} tensorrt_engine pool_op tensorrt_plugin SERIAL )
22
22
nv_test (test_trt_elementwise_op SRCS test_elementwise_op.cc elementwise_op.cc
23
23
DEPS ${FLUID_CORE_MODULES} ${GLOB_OPERATOR_DEPS} tensorrt_engine tensorrt_plugin
24
24
elementwise_add_op elementwise_mul_op SERIAL )
You can’t perform that action at this time.
0 commit comments