mpi/src/starpu_mpi_collective.c: free callback_arg

mpi/src/starpu_mpi_collective.c

@@ -34,6 +34,7 @@ void _callback_collective(void *arg)
 	if (callback_arg->nb == callback_arg->count)
 	{
 		callback_arg->callback(callback_arg->arg);
+		free(callback_arg);
 	}
 }
 
@@ -46,9 +47,6 @@ int starpu_mpi_scatter_detached(starpu_data_handle_t *data_handles, int count, i
 
 	MPI_Comm_rank(comm, &rank);
 
-#ifdef STARPU_DEVEL
-#warning TODO: callback_arg needs to be free-ed
-#endif
 	callback_func = _callback_collective;
 	callback_arg = malloc(sizeof(struct _callback_arg));
 	callback_arg->count = 0;