Explorar o código

synchronize only the transfer stream

Samuel Thibault %!s(int64=13) %!d(string=hai) anos
pai
achega
f18180c477
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      src/datawizard/interfaces/matrix_interface.c

+ 1 - 1
src/datawizard/interfaces/matrix_interface.c

@@ -513,7 +513,7 @@ static int copy_cuda_peer(void *src_interface, unsigned src_node STARPU_ATTRIBUT
 	cures = cudaMemcpy3DPeerAsync(&p, stream);
 	if (STARPU_UNLIKELY(cures))
 		STARPU_CUDA_REPORT_ERROR(cures);
-	cudaThreadSynchronize();
+	cudaStreamSynchronize(stream);
 
 	if (is_async)
 	{