Ver código fonte

Use a memcpy operation that relies on a stream

Cédric Augonnet 14 anos atrás
pai
commit
339892a2a6
1 arquivos alterados com 2 adições e 2 exclusões
  1. 2 2
      examples/cg/cg_kernels.c

+ 2 - 2
examples/cg/cg_kernels.c

@@ -582,8 +582,8 @@ static void copy_handle_cuda(void *descr[], void *cl_arg)
 	unsigned nx = STARPU_VECTOR_GET_NX(descr[0]);
 	size_t elemsize = STARPU_VECTOR_GET_ELEMSIZE(descr[0]);
 
-	cudaMemcpy(dst, src, nx*elemsize, cudaMemcpyDeviceToDevice);
-	cudaThreadSynchronize();
+	cudaMemcpyAsync(dst, src, nx*elemsize, cudaMemcpyDeviceToDevice, starpu_cuda_get_local_stream());
+	cudaStreamSynchronize(starpu_cuda_get_local_stream());
 }
 #endif