浏览代码

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)
 	{