copy data back to gpu

mrirecon · Oct 17, 2023 · dd88d07 · dd88d07
1 parent 30cb5a9
commit dd88d07
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/num/gpuops.c b/src/num/gpuops.c
@@ -310,6 +310,7 @@ static void* cuda_malloc_wrapper(size_t size)
 		if(0 != access) {
 
 			CUDA_ERROR(cudaMemAdvise(ptr, size, cudaMemAdviseSetAccessedBy, cuda_device_id));
+			CUDA_ERROR(cudaMemAdvise(ptr, size, cudaMemAdviseSetPreferredLocation, cuda_device_id));
 			CUDA_ERROR(cudaMemPrefetchAsync(ptr, size, cuda_device_id, cuda_get_stream()));
 		}