浏览代码

examples/pipeline/pipeline.c: synchronise before ending the codelet

Nathalie Furmento 11 年之前
父节点
当前提交
e34466743f
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1 0
      examples/pipeline/pipeline.c

+ 1 - 0
examples/pipeline/pipeline.c

@@ -101,6 +101,7 @@ void pipeline_cublas_axpy(void *descr[], void *arg)
 	int n = STARPU_VECTOR_GET_NX(descr[0]);
 
 	cublasSaxpy(n, 1., x, 1, y, 1);
+	cudaStreamSynchronize(starpu_cuda_get_local_stream());
 }
 #endif