Browse Source

replace malloc with STARPU_MALLOC to check the pointer

Corentin Salingue 8 years ago
parent
commit
6d11b2397f

+ 2 - 2
src/drivers/mic/driver_mic_source.c

@@ -128,14 +128,14 @@ int _starpu_mic_src_register_kernel(starpu_mic_func_symbol_t *symbol, const char
 		return 0;
 	}
 
-	kernel = malloc(sizeof(*kernel));
+	_STARPU_MALLOC(kernel, sizeof(*kernel));
 	if (kernel == NULL)
 	{
 		STARPU_PTHREAD_MUTEX_UNLOCK(&htbl_mutex);
 		return -ENOMEM;
 	}
 
-	kernel->name = malloc(func_name_size);
+	_STARPU_MALLOC(kernel->name, func_name_size);
 	if (kernel->name == NULL)
 	{
 		STARPU_PTHREAD_MUTEX_UNLOCK(&htbl_mutex);

+ 2 - 1
src/drivers/mp_common/source_common.c

@@ -927,7 +927,8 @@ void _starpu_src_common_workers_set(struct _starpu_worker_set * worker_set,
             offsetmemnode[device+1] += worker_set[device].nworkers;
     }
 
-	struct starpu_task **tasks = malloc(sizeof(struct starpu_task *)*nbworkers);
+	struct starpu_task **tasks;
+    _STARPU_MALLOC(tasks, sizeof(struct starpu_task *)*nbworkers);
 
     for (device = 0; device < ndevices; device++)
     {

+ 2 - 1
src/drivers/mpi/driver_mpi_common.c

@@ -423,7 +423,8 @@ void _starpu_mpi_common_measure_bandwidth_latency(double * bandwidth_htod, doubl
     MPI_Comm_rank(MPI_COMM_WORLD, &id_proc);
     MPI_Comm_size(MPI_COMM_WORLD, &nb_proc);
 
-    char * buf = malloc(SIZE_BANDWIDTH);
+    char * buf;
+    _STARPU_MALLOC(buf, SIZE_BANDWIDTH);
     memset(buf, 0, SIZE_BANDWIDTH);
 
     unsigned node;

+ 2 - 2
src/drivers/mpi/driver_mpi_sink.c

@@ -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;
 

+ 2 - 2
src/drivers/mpi/driver_mpi_source.c

@@ -197,14 +197,14 @@ int _starpu_mpi_ms_src_register_kernel(starpu_mpi_ms_func_symbol_t *symbol, cons
 		return 0;
 	}
 
-	kernel = malloc(sizeof(*kernel));
+	_STARPU_MALLOC(kernel, sizeof(*kernel));
 	if (kernel == NULL)
 	{
 		STARPU_PTHREAD_MUTEX_UNLOCK(&htbl_mutex);
 		return -ENOMEM;
 	}
 
-	kernel->name = malloc(func_name_size);
+	_STARPU_MALLOC(kernel->name, func_name_size);
 	if (kernel->name == NULL)
 	{
 		STARPU_PTHREAD_MUTEX_UNLOCK(&htbl_mutex);