fix ci

PaddlePaddle · Dec 10, 2017 · 57f383c · 57f383c
1 parent 37139cf
commit 57f383c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/paddle/operators/math/im2col.cu b/paddle/operators/math/im2col.cu
@@ -160,7 +160,7 @@ template <class T>
 class Col2ImFunctor<paddle::operators::math::ColFormat::kCFO,
                     platform::CUDADeviceContext, T> {
  public:
-  void operator()(const platform::DeviceContext& context,
+  void operator()(const platform::CUDADeviceContext& context,
                   const framework::Tensor& col,
                   const std::vector<int>& dilation,
                   const std::vector<int>& stride,

diff --git a/paddle/operators/math/sequence2batch.cu b/paddle/operators/math/sequence2batch.cu
@@ -41,7 +41,7 @@ __global__ void CopyMatrixRowsKernel(const T* src, T* dst, const size_t* index,
 template <typename T>
 class CopyMatrixRowsFunctor<platform::CUDADeviceContext, T> {
  public:
-  void operator()(const platform::CPUDeviceContext& context,
+  void operator()(const platform::CUDADeviceContext& context,
                   const framework::Tensor& src, const size_t* index,
                   framework::Tensor& dst, bool is_src_index) {
     auto src_dims = src.dims();

diff --git a/paddle/platform/device_context_test.cc b/paddle/platform/device_context_test.cc
@@ -24,7 +24,7 @@ TEST(Device, Init) {
   for (int i = 0; i < count; i++) {
     DeviceContext* device_context = new CUDADeviceContext(GPUPlace(i));
     Eigen::GpuDevice* gpu_device =
-        device_context->template GetEigenDevice<GPUPlace>();
+        device_context->template device_context<CUDADeviceContext>();
     ASSERT_NE(nullptr, gpu_device);
     delete device_context;
   }