提交 e5004f3c 编写于 作者: D Dun Liang

fix ci && test=develop

上级 0c5c561b
...@@ -85,6 +85,10 @@ void BufferedReader::ReadAsync(size_t i) { ...@@ -85,6 +85,10 @@ void BufferedReader::ReadAsync(size_t i) {
memory::Copy(boost::get<platform::CUDAPlace>(place_), gpu_ptr, memory::Copy(boost::get<platform::CUDAPlace>(place_), gpu_ptr,
boost::get<platform::CUDAPinnedPlace>(cpu_place), boost::get<platform::CUDAPinnedPlace>(cpu_place),
cpu_ptr, size, stream); 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 else
// if cpu place is not pinned, async copy is slower than sync copy, // if cpu place is not pinned, async copy is slower than sync copy,
// so we use sync copy instead. // so we use sync copy instead.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册