From e5004f3c1c142b39b12bc3c88faa22acee859efe Mon Sep 17 00:00:00 2001 From: Dun Liang Date: Sun, 20 Jan 2019 16:52:38 +0800 Subject: [PATCH] fix ci && test=develop --- paddle/fluid/operators/reader/buffered_reader.cc | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/paddle/fluid/operators/reader/buffered_reader.cc b/paddle/fluid/operators/reader/buffered_reader.cc index d5a7c50d95c..971db8b37d8 100644 --- 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(place_), gpu_ptr, boost::get(cpu_place), cpu_ptr, size, stream); + else if ((platform::is_gpu_place(cpu_place))) + memory::Copy(boost::get(place_), gpu_ptr, + boost::get(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. -- GitLab