소스 검색

synchronize only the transfer stream

Samuel Thibault 13 년 전
부모
커밋
f18180c477
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  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)
 	{