(hotfix)

szcompressor · Jun 14, 2023 · bd6e8f0 · bd6e8f0
1 parent cd72e15
commit bd6e8f0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/utils2/memseg.cc b/src/utils2/memseg.cc
@@ -163,12 +163,12 @@ void pszmem_d2h_cudaasync(Mem* m, void* stream)
 
 void pszmem_device_deepcpy_cuda(Mem* dst, Mem* src)
 {
-  cudaMemcpy(src->h, dst->d, dst->bytes, cudaMemcpyDeviceToDevice);
+  cudaMemcpy(dst->h, src->d, src->bytes, cudaMemcpyDeviceToDevice);
 }
 
 void pszmem_host_deepcpy_cuda(Mem* dst, Mem* src)
 {
-  cudaMemcpy(src->h, dst->d, dst->bytes, cudaMemcpyHostToHost);
+  cudaMemcpy(dst->h, src->d, src->bytes, cudaMemcpyHostToHost);
 }
 
 void pszmem_fromfile(const char* fname, Mem* m)