Pārlūkot izejas kodu

Make sure we wait for all running tasks before unregistering data

Samuel Thibault 7 gadi atpakaļ
vecāks
revīzija
e2e7f733bb
1 mainītis faili ar 2 papildinājumiem un 0 dzēšanām
  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);