ForkMaster/netgen
1
0
Fork 0
mirror of https://github.com/NGSolve/netgen.git synced 2025-01-12 14:10:34 +05:00
Code Issues Packages Projects Releases Wiki Activity

wrapper for mpi-gather to ngscore

Browse Source
This commit is contained in:
Joachim Schoeberl 2022-03-18 08:20:20 +01:00
parent a5acada9fc
commit 4bad16744b
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
libsrc/core/mpi_wrapper.hpp
View File

@ -303,6 +303,24 @@ namespace ngcore
&recv, 1, GetMPIType<T>(), 0, comm); &recv, 1, GetMPIType<T>(), 0, comm);
} }
template <typename T>
void GatherRoot (FlatArray<T> recv) const
{
recv[0] = T(0);
if (size == 1) return;
MPI_Gather (MPI_IN_PLACE, 1, GetMPIType<T>(),
recv.Data(), 1, GetMPIType<T>(), 0, comm);
}
template <typename T>
void Gather (T send) const
{
if (size == 1) return;
MPI_Gather (&send, 1, GetMPIType<T>(),
NULL, 1, GetMPIType<T>(), 0, comm);
}
template <typename T> template <typename T>
void AllGather (T val, FlatArray<T> recv) const void AllGather (T val, FlatArray<T> recv) const
{ {