|
@@ -27,7 +27,7 @@ void _starpu_mpi_sink_init(struct _starpu_mp_node *node)
|
|
|
{
|
|
|
_starpu_mpi_common_mp_initialize_src_sink(node);
|
|
|
|
|
|
- node->thread_table = malloc(sizeof(starpu_pthread_t)*node->nb_cores);
|
|
|
+ _STARPU_MALLOC(node->thread_table, sizeof(starpu_pthread_t)*node->nb_cores);
|
|
|
//TODO
|
|
|
}
|
|
|
|
|
@@ -64,7 +64,7 @@ void _starpu_mpi_sink_launch_workers(struct _starpu_mp_node *node)
|
|
|
STARPU_ASSERT(ret == 0);
|
|
|
|
|
|
/*prepare the argument for the thread*/
|
|
|
- arg= malloc(sizeof(struct arg_sink_thread));
|
|
|
+ _STARPU_MALLOC(arg, sizeof(struct arg_sink_thread));
|
|
|
arg->coreid = i;
|
|
|
arg->node = node;
|
|
|
|