fix ci && test=develop

zhengqiwen1997 · Jan 20, 2019 · e5004f3 · e5004f3
1 parent 0c5c561
commit e5004f3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/paddle/fluid/operators/reader/buffered_reader.cc b/paddle/fluid/operators/reader/buffered_reader.cc
@@ -85,6 +85,10 @@ void BufferedReader::ReadAsync(size_t i) {
           memory::Copy(boost::get<platform::CUDAPlace>(place_), gpu_ptr,
                        boost::get<platform::CUDAPinnedPlace>(cpu_place),
                        cpu_ptr, size, stream);
+        else if ((platform::is_gpu_place(cpu_place)))
+          memory::Copy(boost::get<platform::CUDAPlace>(place_), gpu_ptr,
+                       boost::get<platform::CUDAPlace>(cpu_place), cpu_ptr,
+                       size, stream);
         else
           // if cpu place is not pinned, async copy is slower than sync copy,
           // so we use sync copy instead.