Преглед на файлове

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