Browse Source

mpi: Fix load balancer interface allocation

Samuel Thibault 4 years ago
parent
commit
3c59078516
1 changed files with 1 additions and 2 deletions
  1. 1 2
      mpi/src/load_balancer/policy/load_data_interface.c

+ 1 - 2
mpi/src/load_balancer/policy/load_data_interface.c

@@ -191,8 +191,7 @@ static int load_data_pack_data(starpu_data_handle_t handle, unsigned node, void
 	*count = load_data_get_size(handle);
 	if (ptr != NULL)
 	{
-		char *data;
-		starpu_malloc_flags((void**) &data, *count, 0);
+		char *data = (void*) starpu_malloc_on_node_flags(node, *count, 0);
 		*ptr = data;
 		memcpy(data, ld_interface, *count);
 	}