Browse Source

src/util/starpu_insert_task_utils.c: free allocated memory

Nathalie Furmento 13 years ago
parent
commit
af806fdba9
1 changed files with 5 additions and 0 deletions
  1. 5 0
      src/util/starpu_insert_task_utils.c

+ 5 - 0
src/util/starpu_insert_task_utils.c

@@ -41,6 +41,7 @@ void starpu_task_insert_callback_wrapper(void *_cl_arg_wrapper)
 
 	/* Free the stack of arguments */
 	free(cl_arg_wrapper->arg_stack);
+	free(cl_arg_wrapper);
 }
 
 size_t _starpu_insert_task_get_arg_size(va_list varg_list)
@@ -250,10 +251,14 @@ int _starpu_insert_task_create_and_submit(char *arg_buffer, struct starpu_codele
 	int ret = starpu_task_submit(*task);
 
 	if (STARPU_UNLIKELY(ret == -ENODEV))
+	{
 		fprintf(stderr, "submission of task %p wih codelet %p failed (symbol `%s')\n",
 			*task, (*task)->cl,
 			(*task)->cl->name ? (*task)->cl->name :
 			((*task)->cl->model && (*task)->cl->model->symbol)?(*task)->cl->model->symbol:"none");
+		free(cl_arg_wrapper->arg_stack);
+		free(cl_arg_wrapper);
+	}
 
         return ret;
 }