Fix H2D and D2H order

527c8597 · liaogang · b0588641 · 527c8597
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

paddle/memory/memory.cc paddle/memory/memory.cc +2 -2

未找到文件。
--- 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 <>