From 8f5a9fd9a7297007dc259114c23d986c8ee4e06a Mon Sep 17 00:00:00 2001
From: qijun <qijun@baidu.com>
Date: Wed, 12 Jul 2017 11:20:14 +0800
Subject: [PATCH] fix gpu build error

---
 paddle/platform/CMakeLists.txt        | 2 +-
 paddle/platform/cuda_device_context.h | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/paddle/platform/CMakeLists.txt b/paddle/platform/CMakeLists.txt
index 4e34e8d02..5e2f20355 100644
--- 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)
diff --git a/paddle/platform/cuda_device_context.h b/paddle/platform/cuda_device_context.h
index e0d79631c..0ba1f802a 100644
--- 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 {
-- 
GitLab