Bladeren bron

mpi_pxlu: Fix unregistering data

Samuel Thibault 4 jaren geleden
bovenliggende
commit
8efc8b2c29
1 gewijzigde bestanden met toevoegingen van 3 en 1 verwijderingen
  1. 3 1
      mpi/examples/mpi_lu/plu_example.c

+ 3 - 1
mpi/examples/mpi_lu/plu_example.c

@@ -610,7 +610,9 @@ int main(int argc, char **argv)
 	{
 		for (i = 0; i < nblocks; i++)
 		{
-			starpu_data_unregister(dataA_handles[j+nblocks*i]);
+			starpu_data_handle_t handle = dataA_handles[j+nblocks*i];
+			if (handle != STARPU_POISON_PTR)
+				starpu_data_unregister(handle);
 			TYPE *blockptr = dataA[j+i*nblocks];
 			if (blockptr != STARPU_POISON_PTR)
 				starpu_free(blockptr);