fix gpu build error

8 years ago · 8f5a9fd9a7
parent 14d2c3990f
commit 8f5a9fd9a7
2 changed files with 2 additions and 1 deletions
--- a/paddle/platform/CMakeLists.txt
+++ b/paddle/platform/CMakeLists.txt
@ -5,4 +5,4 @@ nv_test(cuda_test SRCS cuda_test.cu DEPS dynload_cuda)
 cc_library(place SRCS place.cc)
 cc_test(place_test SRCS place_test.cc DEPS place glog gflags)

-nv_test(device_context_test SRCS device_context_test.cc DEPS place eigen3 dynload_cuda)
+nv_test(device_context_test SRCS device_context_test.cc DEPS dynload_cuda dynamic_loader eigen3 place)
--- a/paddle/platform/cuda_device_context.h
+++ b/paddle/platform/cuda_device_context.h
@ -21,6 +21,7 @@ limitations under the License. */
 #include "paddle/platform/dynload/curand.h"
 #define EIGEN_USE_GPU
 #include "paddle/platform/place.h"
+#include "paddle/platform/device_context.h"
 #include "unsupported/Eigen/CXX11/Tensor"

 namespace paddle {