rerun ci

7 years ago · 37842d802d
parent 75eacccd5c
commit 37842d802d
1 changed files with 2 additions and 0 deletions
--- a/paddle/pybind/pybind.cc
+++ b/paddle/pybind/pybind.cc
@ -483,7 +483,9 @@ All parameter, weight, gradient are variables in Paddle.
  BindOpDesc(m);

  m.def("op_support_gpu", OpSupportGPU);
+#ifdef PADDLE_WITH_CUDA
  m.def("get_cuda_device_count", platform::GetCUDADeviceCount);
+#endif

  return m.ptr();
 }