Fix H2D and D2H order

8 years ago · 527c85970b
parent b05886414b
commit 527c85970b
1 changed files with 2 additions and 2 deletions
--- a/paddle/memory/memory.cc
+++ b/paddle/memory/memory.cc
@ -88,7 +88,7 @@ void Copy<platform::CPUPlace, platform::GPUPlace>(platform::CPUPlace, void* dst,
                                                  platform::GPUPlace,
                                                  const void* src, size_t num,
                                                  cudaStream_t stream) {
-  platform::GpuMemcpyAsync(dst, src, num, cudaMemcpyHostToDevice, stream);
+  platform::GpuMemcpyAsync(dst, src, num, cudaMemcpyDeviceToHost, stream);
 }

 template <>
@ -96,7 +96,7 @@ void Copy<platform::GPUPlace, platform::CPUPlace>(platform::GPUPlace, void* dst,
                                                  platform::CPUPlace,
                                                  const void* src, size_t num,
                                                  cudaStream_t stream) {
-  platform::GpuMemcpyAsync(dst, src, num, cudaMemcpyDeviceToHost, stream);
+  platform::GpuMemcpyAsync(dst, src, num, cudaMemcpyHostToDevice, stream);
 }

 template <>