Преглед изворни кода

mpi: Fix load balancer interface allocation

Samuel Thibault пре 4 година
родитељ
комит
3c59078516
1 измењених фајлова са 1 додато и 2 уклоњено
  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);
 	}