diff --git a/paddle/pybind/pybind.cc b/paddle/pybind/pybind.cc index e1e382b2bbc9c892c6bffee353464680eec7d06a..9288468a0381c5dbcf54255bef6e97aa53f510c4 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() {