fix ci && test=develop

6 years ago · e5004f3c1c
parent 0c5c561bd1
commit e5004f3c1c
1 changed files with 4 additions and 0 deletions
--- 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.