From 48d7b702bcffae85069cf806e0813124af6f4d07 Mon Sep 17 00:00:00 2001 From: superjomn <yanchunwei@outlook.com> Date: Tue, 29 May 2018 17:00:15 +0800 Subject: [PATCH] add if --- paddle/fluid/operators/CMakeLists.txt | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/operators/CMakeLists.txt b/paddle/fluid/operators/CMakeLists.txt index c4299191a0385..6e8f5aa52b8c5 100644 --- a/paddle/fluid/operators/CMakeLists.txt +++ b/paddle/fluid/operators/CMakeLists.txt @@ -221,7 +221,9 @@ op_library(cross_entropy_op DEPS cross_entropy) op_library(softmax_with_cross_entropy_op DEPS cross_entropy softmax) op_library(softmax_op DEPS softmax) op_library(sequence_softmax_op DEPS softmax) -op_library(tensorrt_engine_op DEPS tensorrt_engine) +if (TENSORRT_FOUND) + op_library(tensorrt_engine_op DEPS tensorrt_engine) +endif() op_library(sum_op DEPS selected_rows_functor) op_library(sgd_op DEPS selected_rows_functor) op_library(print_op DEPS lod_tensor)