Преглед изворни кода

Fix memleak when count is 0

Samuel Thibault пре 10 година
родитељ
комит
256bc993b3
1 измењених фајлова са 6 додато и 0 уклоњено
  1. 6 0
      mpi/src/starpu_mpi_collective.c

+ 6 - 0
mpi/src/starpu_mpi_collective.c

@@ -140,6 +140,12 @@ int starpu_mpi_gather_detached(starpu_data_handle_t *data_handles, int count, in
 		}
 	}
 
+	if (!callback_arg->count)
+	{
+		free(callback_arg);
+		return 0;
+	}
+
 	for(x = 0; x < count ; x++)
 	{
 		if (data_handles[x])