10 #include <dolfinx/common/IndexMap.h>
11 #include <dolfinx/common/Timer.h>
12 #include <dolfinx/graph/AdjacencyList.h>
49 static std::pair<std::vector<std::int32_t>, std::vector<std::int64_t>>
51 const std::vector<std::int64_t>& global_indices,
52 const std::vector<bool>& shared_indices);
62 static std::pair<graph::AdjacencyList<std::int32_t>,
63 std::vector<std::int64_t>>
79 const std::vector<std::int64_t>& local_to_global_links,
80 const std::vector<bool>& shared_links);
93 static std::tuple<graph::AdjacencyList<std::int64_t>, std::vector<int>,
94 std::vector<std::int64_t>, std::vector<int>>
105 static std::vector<std::int64_t>
107 const std::vector<std::int64_t>& global_indices,
108 const std::vector<int>& ghost_owners);
119 template <
typename T>
120 static Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor>
121 distribute_data(MPI_Comm comm,
const std::vector<std::int64_t>& indices,
122 const Eigen::Ref<
const Eigen::Array<
123 T, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor>>& x);
148 static std::vector<std::int32_t>
150 const std::vector<std::int64_t>& local1_to_global);
156 template <
typename T>
157 Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor>
159 MPI_Comm comm,
const std::vector<std::int64_t>& indices,
160 const Eigen::Ref<
const Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic,
161 Eigen::RowMajor>>& x)
163 common::Timer timer(
"Fetch float data from remote processes");
165 const std::int64_t num_points_local = x.rows();
168 std::vector<std::int64_t> global_sizes(size);
169 MPI_Allgather(&num_points_local, 1, MPI_INT64_T, global_sizes.data(), 1,
171 std::vector<std::int64_t> global_offsets(size + 1, 0);
172 std::partial_sum(global_sizes.begin(), global_sizes.end(),
173 global_offsets.begin() + 1);
176 std::vector<int> number_index_send(size, 0);
177 std::vector<int> index_owner(indices.size());
178 std::vector<int> index_order(indices.size());
179 std::iota(index_order.begin(), index_order.end(), 0);
180 std::sort(index_order.begin(), index_order.end(),
181 [&indices](
int a,
int b) { return (indices[a] < indices[b]); });
184 for (std::size_t i = 0; i < index_order.size(); ++i)
186 int j = index_order[i];
187 while (indices[j] >= global_offsets[p + 1])
190 number_index_send[p]++;
194 std::vector<int> disp_index_send(size + 1, 0);
195 std::partial_sum(number_index_send.begin(), number_index_send.end(),
196 disp_index_send.begin() + 1);
199 std::vector<std::int64_t> indices_send(disp_index_send.back());
200 std::vector<int> disp_tmp = disp_index_send;
201 for (std::size_t i = 0; i < indices.size(); ++i)
203 const int owner = index_owner[i];
204 indices_send[disp_tmp[owner]++] = indices[i];
208 std::vector<int> number_index_recv(size);
209 MPI_Alltoall(number_index_send.data(), 1, MPI_INT, number_index_recv.data(),
213 std::vector<int> disp_index_recv(size + 1, 0);
214 std::partial_sum(number_index_recv.begin(), number_index_recv.end(),
215 disp_index_recv.begin() + 1);
218 std::vector<std::int64_t> indices_recv(disp_index_recv.back());
219 MPI_Alltoallv(indices_send.data(), number_index_send.data(),
220 disp_index_send.data(), MPI_INT64_T, indices_recv.data(),
221 number_index_recv.data(), disp_index_recv.data(), MPI_INT64_T,
224 const int item_size = x.cols();
225 assert(item_size != 0);
227 Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> x_return(
228 indices_recv.size(), item_size);
229 for (
int p = 0; p < size; ++p)
231 for (
int i = disp_index_recv[p]; i < disp_index_recv[p + 1]; ++i)
233 const std::int32_t index_local = indices_recv[i] - global_offsets[rank];
234 assert(index_local >= 0);
235 x_return.row(i) = x.row(index_local);
239 MPI_Datatype compound_type;
240 MPI_Type_contiguous(item_size, dolfinx::MPI::mpi_type<T>(), &compound_type);
241 MPI_Type_commit(&compound_type);
244 Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor> my_x(
245 disp_index_send.back(), item_size);
246 MPI_Alltoallv(x_return.data(), number_index_recv.data(),
247 disp_index_recv.data(), compound_type, my_x.data(),
248 number_index_send.data(), disp_index_send.data(), compound_type,