소스 검색

mpi: Fix load balancer interface allocation

Samuel Thibault 5 년 전
부모
커밋
b97f4859cb
1개의 변경된 파일1개의 추가작업 그리고 2개의 파일을 삭제
  1. 1 2
      mpi/src/load_balancer/policy/data_movements_interface.c

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

@@ -187,8 +187,7 @@ static int data_movements_pack_data(starpu_data_handle_t handle, unsigned node,
 	*count = data_movements_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);
 		assert(data);
 		*ptr = data;
 		memcpy(data, &dm_interface->size, sizeof(int));