add if

PaddlePaddle · May 29, 2018 · 48d7b70 · 48d7b70
1 parent c4e29a5
commit 48d7b70
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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)