Browse Source

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

Nathalie Furmento 11 years ago
parent
commit
e34466743f
1 changed files with 1 additions and 0 deletions
  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