From dbfd1302e1486939b33b79b2485b0889f5cc2994 Mon Sep 17 00:00:00 2001 From: Dong Zhihong Date: Thu, 26 Oct 2017 10:57:52 -0700 Subject: [PATCH] "FIX CI" --- paddle/pybind/pybind.cc | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/paddle/pybind/pybind.cc b/paddle/pybind/pybind.cc index e1e382b2b..9288468a0 100644 --- a/paddle/pybind/pybind.cc +++ b/paddle/pybind/pybind.cc @@ -23,7 +23,6 @@ limitations under the License. */ #include "paddle/framework/tensor_array.h" #include "paddle/operators/cond_op.h" #include "paddle/operators/dynamic_recurrent_op.h" -#include "paddle/operators/nccl/nccl_gpu_common.h" #include "paddle/operators/net_op.h" #include "paddle/operators/recurrent_op.h" #include "paddle/platform/enforce.h" @@ -33,6 +32,10 @@ limitations under the License. */ #include "paddle/pybind/tensor_py.h" #include "paddle/string/to_string.h" +#ifdef PADDLE_WITH_CUDA +#include "paddle/operators/nccl/nccl_gpu_common.h" +#endif + namespace paddle { namespace pybind { static size_t UniqueIntegerGenerator() { -- GitLab