浏览代码

Make sure we wait for all running tasks before unregistering data

Samuel Thibault 7 年之前
父节点
当前提交
e2e7f733bb
共有 1 个文件被更改,包括 2 次插入0 次删除
  1. 2 0
      examples/heat/dw_sparse_cg.c

+ 2 - 0
examples/heat/dw_sparse_cg.c

@@ -448,6 +448,8 @@ void conjugate_gradient(float *nzvalA, float *vecb, float *vecx, uint32_t nnz,
 	sem_wait(&sem);
 	sem_destroy(&sem);
 
+	starpu_task_wait_for_all();
+
 	print_results(vecx, nrow);
 
 	starpu_data_unregister(ds_matrixA);