From 514648665ae7093f4531579246e5fec7c1849f07 Mon Sep 17 00:00:00 2001 From: nhzlx Date: Sat, 15 Dec 2018 08:01:27 +0000 Subject: [PATCH] fix trt_op test test=develop --- paddle/fluid/operators/tensorrt/tensorrt_engine_op_test.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/paddle/fluid/operators/tensorrt/tensorrt_engine_op_test.cc b/paddle/fluid/operators/tensorrt/tensorrt_engine_op_test.cc index 6f8adb00e..287b0edc9 100644 --- a/paddle/fluid/operators/tensorrt/tensorrt_engine_op_test.cc +++ b/paddle/fluid/operators/tensorrt/tensorrt_engine_op_test.cc @@ -24,6 +24,7 @@ limitations under the License. */ #include "paddle/fluid/inference/tensorrt/convert/op_converter.h" #include "paddle/fluid/inference/tensorrt/convert/ut_helper.h" +USE_NO_KERNEL_OP(tensorrt_engine); namespace paddle { namespace operators { -- GitLab