Nektar++
Public Member Functions | Static Public Member Functions | Static Public Attributes | Protected Member Functions | Protected Attributes | List of all members
Nektar::LibUtilities::CommMpi Class Reference

A global linear system. More...

#include <CommMpi.h>

Inheritance diagram for Nektar::LibUtilities::CommMpi:
[legend]

Public Member Functions

 CommMpi (int narg, char *arg[])
 
 ~CommMpi () override
 
MPI_Comm GetComm ()
 
- Public Member Functions inherited from Nektar::LibUtilities::Comm
 Comm (int narg, char *arg[])
 
virtual ~Comm ()
 
void Finalise ()
 
int GetSize () const
 Returns number of processes. More...
 
int GetRank ()
 
const std::string & GetType () const
 
bool TreatAsRankZero ()
 
bool IsSerial ()
 
bool IsParallelInTime ()
 
std::tuple< int, int, int > GetVersion ()
 
void Block ()
 Block execution until all processes reach this point. More...
 
NekDouble Wtime ()
 Return the time in seconds. More...
 
template<class T >
void Send (int pProc, T &pData)
 
template<class T >
void Recv (int pProc, T &pData)
 
template<class T >
void SendRecv (int pSendProc, T &pSendData, int pRecvProc, T &pRecvData)
 
template<class T >
void AllReduce (T &pData, enum ReduceOperator pOp)
 
template<class T >
void AlltoAll (T &pSendData, T &pRecvData)
 
template<class T1 , class T2 >
void AlltoAllv (T1 &pSendData, T2 &pSendDataSizeMap, T2 &pSendDataOffsetMap, T1 &pRecvData, T2 &pRecvDataSizeMap, T2 &pRecvDataOffsetMap)
 
template<class T >
void AllGather (T &pSendData, T &pRecvData)
 
template<class T >
void AllGatherv (T &pSendData, T &pRecvData, Array< OneD, int > &pRecvDataSizeMap, Array< OneD, int > &pRecvDataOffsetMap)
 
template<class T >
void AllGatherv (T &pRecvData, Array< OneD, int > &pRecvDataSizeMap, Array< OneD, int > &pRecvDataOffsetMap)
 
template<class T >
void Bcast (T &pData, int pRoot)
 
template<class T >
Gather (int rootProc, T &val)
 
template<class T >
Scatter (int rootProc, T &pData)
 
template<class T >
void DistGraphCreateAdjacent (T &sources, T &sourceweights, int reorder)
 
template<class T1 , class T2 >
void NeighborAlltoAllv (T1 &pSendData, T2 &pSendDataSizeMap, T2 &pSendDataOffsetMap, T1 &pRecvData, T2 &pRecvDataSizeMap, T2 &pRecvDataOffsetMap)
 
template<class T >
void Irsend (int pProc, T &pData, int count, const CommRequestSharedPtr &request, int loc)
 
template<class T >
void Isend (int pProc, T &pData, int count, const CommRequestSharedPtr &request, int loc)
 
template<class T >
void SendInit (int pProc, T &pData, int count, const CommRequestSharedPtr &request, int loc)
 
template<class T >
void Irecv (int pProc, T &pData, int count, const CommRequestSharedPtr &request, int loc)
 
template<class T >
void RecvInit (int pProc, T &pData, int count, const CommRequestSharedPtr &request, int loc)
 
void StartAll (const CommRequestSharedPtr &request)
 
void WaitAll (const CommRequestSharedPtr &request)
 
CommRequestSharedPtr CreateRequest (int num)
 
CommSharedPtr CommCreateIf (int flag)
 If the flag is non-zero create a new communicator. More...
 
void SplitComm (int pRows, int pColumns, int pTime=1)
 Splits this communicator into a grid of size pRows*pColumns and creates row and column communicators. By default the communicator is a single row. More...
 
CommSharedPtr GetRowComm ()
 Retrieve the row communicator to which this process belongs. More...
 
CommSharedPtr GetColumnComm ()
 Retrieve the column communicator to which this process belongs. More...
 
CommSharedPtr GetTimeComm ()
 Retrieve the time communicator to which this process belongs. More...
 
CommSharedPtr GetSpaceComm ()
 Retrieve the space communicator to which this process belongs. More...
 
bool RemoveExistingFiles ()
 
std::pair< CommSharedPtr, CommSharedPtrSplitCommNode ()
 

Static Public Member Functions

static CommSharedPtr create (int narg, char *arg[])
 Creates an instance of this class. More...
 

Static Public Attributes

static std::string className
 Name of class. More...
 

Protected Member Functions

 CommMpi (MPI_Comm pComm)
 
void v_Finalise () override
 
int v_GetRank () final
 
bool v_TreatAsRankZero () final
 
bool v_IsSerial () final
 
std::tuple< int, int, int > v_GetVersion () final
 
void v_Block () final
 
double v_Wtime () final
 
void v_Send (void *buf, int count, CommDataType dt, int dest) final
 
void v_Recv (void *buf, int count, CommDataType dt, int source) final
 
void v_SendRecv (void *sendbuf, int sendcount, CommDataType sendtype, int dest, void *recvbuf, int recvcount, CommDataType recvtype, int source) final
 
void v_AllReduce (void *buf, int count, CommDataType dt, enum ReduceOperator pOp) final
 
void v_AlltoAll (void *sendbuf, int sendcount, CommDataType sendtype, void *recvbuf, int recvcount, CommDataType recvtype) final
 
void v_AlltoAllv (void *sendbuf, int sendcounts[], int sensdispls[], CommDataType sendtype, void *recvbuf, int recvcounts[], int rdispls[], CommDataType recvtype) final
 
void v_AllGather (void *sendbuf, int sendcount, CommDataType sendtype, void *recvbuf, int recvcount, CommDataType recvtype) final
 
void v_AllGatherv (void *sendbuf, int sendcount, CommDataType sendtype, void *recvbuf, int recvcounts[], int rdispls[], CommDataType recvtype) final
 
void v_AllGatherv (void *recvbuf, int recvcounts[], int rdispls[], CommDataType recvtype) final
 
void v_Bcast (void *buffer, int count, CommDataType dt, int root) final
 
void v_Gather (void *sendbuf, int sendcount, CommDataType sendtype, void *recvbuf, int recvcount, CommDataType recvtype, int root) final
 
void v_Scatter (void *sendbuf, int sendcount, CommDataType sendtype, void *recvbuf, int recvcount, CommDataType recvtype, int root) final
 
void v_DistGraphCreateAdjacent (int indegree, const int sources[], const int sourceweights[], int reorder) final
 
void v_NeighborAlltoAllv (void *sendbuf, int sendcounts[], int sensdispls[], CommDataType sendtype, void *recvbuf, int recvcounts[], int rdispls[], CommDataType recvtype) final
 
void v_Irsend (void *buf, int count, CommDataType dt, int dest, CommRequestSharedPtr request, int loc) final
 
void v_Isend (void *buf, int count, CommDataType dt, int dest, CommRequestSharedPtr request, int loc) final
 
void v_SendInit (void *buf, int count, CommDataType dt, int dest, CommRequestSharedPtr request, int loc) final
 
void v_Irecv (void *buf, int count, CommDataType dt, int source, CommRequestSharedPtr request, int loc) final
 
void v_RecvInit (void *buf, int count, CommDataType dt, int source, CommRequestSharedPtr request, int loc) final
 
void v_StartAll (CommRequestSharedPtr request) final
 
void v_WaitAll (CommRequestSharedPtr request) final
 
CommRequestSharedPtr v_CreateRequest (int num) final
 
void v_SplitComm (int pRows, int pColumns, int pTime) override
 
CommSharedPtr v_CommCreateIf (int flag) final
 
std::pair< CommSharedPtr, CommSharedPtrv_SplitCommNode () final
 
- Protected Member Functions inherited from Nektar::LibUtilities::Comm
 Comm ()
 
virtual void v_Finalise ()=0
 
virtual int v_GetRank ()=0
 
virtual bool v_TreatAsRankZero ()=0
 
virtual bool v_IsSerial ()=0
 
virtual std::tuple< int, int, int > v_GetVersion ()=0
 
virtual void v_Block ()=0
 
virtual NekDouble v_Wtime ()=0
 
virtual void v_Send (void *buf, int count, CommDataType dt, int dest)=0
 
virtual void v_Recv (void *buf, int count, CommDataType dt, int source)=0
 
virtual void v_SendRecv (void *sendbuf, int sendcount, CommDataType sendtype, int dest, void *recvbuf, int recvcount, CommDataType recvtype, int source)=0
 
virtual void v_AllReduce (void *buf, int count, CommDataType dt, enum ReduceOperator pOp)=0
 
virtual void v_AlltoAll (void *sendbuf, int sendcount, CommDataType sendtype, void *recvbuf, int recvcount, CommDataType recvtype)=0
 
virtual void v_AlltoAllv (void *sendbuf, int sendcounts[], int sensdispls[], CommDataType sendtype, void *recvbuf, int recvcounts[], int rdispls[], CommDataType recvtype)=0
 
virtual void v_AllGather (void *sendbuf, int sendcount, CommDataType sendtype, void *recvbuf, int recvcount, CommDataType recvtype)=0
 
virtual void v_AllGatherv (void *sendbuf, int sendcount, CommDataType sendtype, void *recvbuf, int recvcounts[], int rdispls[], CommDataType recvtype)=0
 
virtual void v_AllGatherv (void *recvbuf, int recvcounts[], int rdispls[], CommDataType recvtype)=0
 
virtual void v_Bcast (void *buffer, int count, CommDataType dt, int root)=0
 
virtual void v_Gather (void *sendbuf, int sendcount, CommDataType sendtype, void *recvbuf, int recvcount, CommDataType recvtype, int root)=0
 
virtual void v_Scatter (void *sendbuf, int sendcount, CommDataType sendtype, void *recvbuf, int recvcount, CommDataType recvtype, int root)=0
 
virtual void v_DistGraphCreateAdjacent (int indegree, const int sources[], const int sourceweights[], int reorder)=0
 
virtual void v_NeighborAlltoAllv (void *sendbuf, int sendcounts[], int sdispls[], CommDataType sendtype, void *recvbuf, int recvcounts[], int rdispls[], CommDataType recvtype)=0
 
virtual void v_Irsend (void *buf, int count, CommDataType dt, int dest, CommRequestSharedPtr request, int loc)=0
 
virtual void v_Isend (void *buf, int count, CommDataType dt, int dest, CommRequestSharedPtr request, int loc)=0
 
virtual void v_SendInit (void *buf, int count, CommDataType dt, int dest, CommRequestSharedPtr request, int loc)=0
 
virtual void v_Irecv (void *buf, int count, CommDataType dt, int source, CommRequestSharedPtr request, int loc)=0
 
virtual void v_RecvInit (void *buf, int count, CommDataType dt, int source, CommRequestSharedPtr request, int loc)=0
 
virtual void v_StartAll (CommRequestSharedPtr request)=0
 
virtual void v_WaitAll (CommRequestSharedPtr request)=0
 
virtual CommRequestSharedPtr v_CreateRequest (int num)=0
 
virtual void v_SplitComm (int pRows, int pColumns, int pTime)=0
 
virtual CommSharedPtr v_CommCreateIf (int flag)=0
 
virtual std::pair< CommSharedPtr, CommSharedPtrv_SplitCommNode ()
 

Protected Attributes

MPI_Comm m_comm
 
int m_rank {}
 
bool m_controls_mpi
 
- Protected Attributes inherited from Nektar::LibUtilities::Comm
int m_size
 Number of processes. More...
 
std::string m_type
 Type of communication. More...
 
CommSharedPtr m_commRow
 Row communicator. More...
 
CommSharedPtr m_commColumn
 Column communicator. More...
 
CommSharedPtr m_commTime
 
CommSharedPtr m_commSpace
 

Detailed Description

A global linear system.

Definition at line 89 of file CommMpi.h.

Constructor & Destructor Documentation

◆ CommMpi() [1/2]

Nektar::LibUtilities::CommMpi::CommMpi ( int  narg,
char *  arg[] 
)

Definition at line 50 of file CommMpi.cpp.

50 : Comm(narg, arg)
51{
52 int init = 0;
53 MPI_Initialized(&init);
54
55 if (!init)
56 {
57 int thread_support = 0;
58 if (MPI_Init_thread(&narg, &arg, MPI_THREAD_MULTIPLE,
59 &thread_support) != MPI_SUCCESS)
60 {
63 "Initializing MPI using MPI_Init, if scotch version > 6 and is "
64 "compiled with multi-threading, it might cause deadlocks.")
65 ASSERTL0(MPI_Init(&narg, &arg) == MPI_SUCCESS,
66 "Failed to initialise MPI");
67 }
68 // store bool to indicate that Nektar++ is in charge of finalizing MPI.
69 m_controls_mpi = true;
70 }
71 else
72 {
73 // Another code is in charge of finalizing MPI and this is not the
74 // responsiblity of Nektar++
75 m_controls_mpi = false;
76 }
77
78 m_comm = MPI_COMM_WORLD;
79 MPI_Comm_size(m_comm, &m_size);
80 MPI_Comm_rank(m_comm, &m_rank);
81
82#ifdef NEKTAR_USING_PETSC
83 PetscInitializeNoArguments();
84#endif
85
86 m_type = "Parallel MPI";
87}
#define ASSERTL0(condition, msg)
Definition: ErrorUtil.hpp:208
#define NEKERROR(type, msg)
Assert Level 0 – Fundamental assert which is used whether in FULLDEBUG, DEBUG or OPT compilation mode...
Definition: ErrorUtil.hpp:202
int m_size
Number of processes.
Definition: Comm.h:175
std::string m_type
Type of communication.
Definition: Comm.h:176

References ASSERTL0, Nektar::ErrorUtil::ewarning, m_comm, m_controls_mpi, m_rank, Nektar::LibUtilities::Comm::m_size, Nektar::LibUtilities::Comm::m_type, and NEKERROR.

Referenced by v_CommCreateIf(), v_SplitComm(), and v_SplitCommNode().

◆ ~CommMpi()

Nektar::LibUtilities::CommMpi::~CommMpi ( )
override

Definition at line 104 of file CommMpi.cpp.

105{
106 int flag;
107 MPI_Finalized(&flag);
108 if (!flag && m_comm != MPI_COMM_WORLD)
109 {
110 MPI_Comm_free(&m_comm);
111 }
112}

References m_comm.

◆ CommMpi() [2/2]

Nektar::LibUtilities::CommMpi::CommMpi ( MPI_Comm  pComm)
explicitprotected

Definition at line 92 of file CommMpi.cpp.

92 : Comm()
93{
94 m_comm = pComm;
95 MPI_Comm_size(m_comm, &m_size);
96 MPI_Comm_rank(m_comm, &m_rank);
97
98 m_type = "Parallel MPI";
99}

References m_comm, m_rank, Nektar::LibUtilities::Comm::m_size, and Nektar::LibUtilities::Comm::m_type.

Member Function Documentation

◆ create()

static CommSharedPtr Nektar::LibUtilities::CommMpi::create ( int  narg,
char *  arg[] 
)
inlinestatic

Creates an instance of this class.

Definition at line 93 of file CommMpi.h.

94 {
96 }
static std::shared_ptr< DataType > AllocateSharedPtr(const Args &...args)
Allocate a shared pointer from the memory pool.

References Nektar::MemoryManager< DataType >::AllocateSharedPtr().

◆ GetComm()

MPI_Comm Nektar::LibUtilities::CommMpi::GetComm ( )

Definition at line 117 of file CommMpi.cpp.

118{
119 return m_comm;
120}

References m_comm.

◆ v_AllGather()

void Nektar::LibUtilities::CommMpi::v_AllGather ( void sendbuf,
int  sendcount,
CommDataType  sendtype,
void recvbuf,
int  recvcount,
CommDataType  recvtype 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 288 of file CommMpi.cpp.

290{
291 int retval = MPI_Allgather(sendbuf, sendcount, sendtype, recvbuf, recvcount,
292 recvtype, m_comm);
293
294 ASSERTL0(retval == MPI_SUCCESS, "MPI error performing Allgather.");
295}

References ASSERTL0, and m_comm.

◆ v_AllGatherv() [1/2]

void Nektar::LibUtilities::CommMpi::v_AllGatherv ( void recvbuf,
int  recvcounts[],
int  rdispls[],
CommDataType  recvtype 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 313 of file CommMpi.cpp.

315{
316 int retval = MPI_Allgatherv(MPI_IN_PLACE, 0, MPI_DATATYPE_NULL, recvbuf,
317 recvcounts, rdispls, recvtype, m_comm);
318
319 ASSERTL0(retval == MPI_SUCCESS, "MPI error performing Allgatherv.");
320}

References ASSERTL0, and m_comm.

◆ v_AllGatherv() [2/2]

void Nektar::LibUtilities::CommMpi::v_AllGatherv ( void sendbuf,
int  sendcount,
CommDataType  sendtype,
void recvbuf,
int  recvcounts[],
int  rdispls[],
CommDataType  recvtype 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 300 of file CommMpi.cpp.

303{
304 int retval = MPI_Allgatherv(sendbuf, sendcount, sendtype, recvbuf,
305 recvcounts, rdispls, recvtype, m_comm);
306
307 ASSERTL0(retval == MPI_SUCCESS, "MPI error performing Allgatherv.");
308}

References ASSERTL0, and m_comm.

◆ v_AllReduce()

void Nektar::LibUtilities::CommMpi::v_AllReduce ( void buf,
int  count,
CommDataType  dt,
enum ReduceOperator  pOp 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 232 of file CommMpi.cpp.

234{
235 if (GetSize() == 1)
236 {
237 return;
238 }
239
240 MPI_Op vOp;
241 switch (pOp)
242 {
243 case ReduceMax:
244 vOp = MPI_MAX;
245 break;
246 case ReduceMin:
247 vOp = MPI_MIN;
248 break;
249 case ReduceSum:
250 default:
251 vOp = MPI_SUM;
252 break;
253 }
254 int retval = MPI_Allreduce(MPI_IN_PLACE, buf, count, dt, vOp, m_comm);
255
256 ASSERTL0(retval == MPI_SUCCESS, "MPI error performing All-reduce.");
257}
int GetSize() const
Returns number of processes.
Definition: Comm.h:264

References ASSERTL0, Nektar::LibUtilities::Comm::GetSize(), m_comm, Nektar::LibUtilities::ReduceMax, Nektar::LibUtilities::ReduceMin, and Nektar::LibUtilities::ReduceSum.

◆ v_AlltoAll()

void Nektar::LibUtilities::CommMpi::v_AlltoAll ( void sendbuf,
int  sendcount,
CommDataType  sendtype,
void recvbuf,
int  recvcount,
CommDataType  recvtype 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 262 of file CommMpi.cpp.

264{
265 int retval = MPI_Alltoall(sendbuf, sendcount, sendtype, recvbuf, recvcount,
266 recvtype, m_comm);
267
268 ASSERTL0(retval == MPI_SUCCESS, "MPI error performing All-to-All.");
269}

References ASSERTL0, and m_comm.

◆ v_AlltoAllv()

void Nektar::LibUtilities::CommMpi::v_AlltoAllv ( void sendbuf,
int  sendcounts[],
int  sensdispls[],
CommDataType  sendtype,
void recvbuf,
int  recvcounts[],
int  rdispls[],
CommDataType  recvtype 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 274 of file CommMpi.cpp.

278{
279 int retval = MPI_Alltoallv(sendbuf, sendcounts, sdispls, sendtype, recvbuf,
280 recvcounts, rdispls, recvtype, m_comm);
281
282 ASSERTL0(retval == MPI_SUCCESS, "MPI error performing All-to-All-v.");
283}

References ASSERTL0, and m_comm.

◆ v_Bcast()

void Nektar::LibUtilities::CommMpi::v_Bcast ( void buffer,
int  count,
CommDataType  dt,
int  root 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 325 of file CommMpi.cpp.

326{
327 int retval = MPI_Bcast(buffer, count, dt, root, m_comm);
328
329 ASSERTL0(retval == MPI_SUCCESS, "MPI error performing Bcast-v.");
330}
array buffer
Definition: GsLib.hpp:81

References ASSERTL0, and m_comm.

◆ v_Block()

void Nektar::LibUtilities::CommMpi::v_Block ( )
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 178 of file CommMpi.cpp.

179{
180 MPI_Barrier(m_comm);
181}

References m_comm.

◆ v_CommCreateIf()

CommSharedPtr Nektar::LibUtilities::CommMpi::v_CommCreateIf ( int  flag)
finalprotectedvirtual

Create a new communicator if the flag is non-zero.

Implements Nektar::LibUtilities::Comm.

Definition at line 550 of file CommMpi.cpp.

551{
552 MPI_Comm newComm;
553 // color == MPI_UNDEF => not in the new communicator
554 // key == 0 on all => use rank to order them. OpenMPI, at least,
555 // implies this is faster than ordering them ourselves.
556 MPI_Comm_split(m_comm, flag ? flag : MPI_UNDEFINED, 0, &newComm);
557
558 if (flag == 0)
559 {
560 // flag == 0 => get back MPI_COMM_NULL, return a null ptr instead.
561 return std::shared_ptr<Comm>();
562 }
563 else
564 {
565 // Return a real communicator
566 return std::shared_ptr<Comm>(new CommMpi(newComm));
567 }
568}
CommMpi(int narg, char *arg[])
Definition: CommMpi.cpp:50

References CommMpi(), and m_comm.

Referenced by v_SplitCommNode().

◆ v_CreateRequest()

CommRequestSharedPtr Nektar::LibUtilities::CommMpi::v_CreateRequest ( int  num)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 484 of file CommMpi.cpp.

485{
486 return std::shared_ptr<CommRequest>(new CommRequestMpi(num));
487}

◆ v_DistGraphCreateAdjacent()

void Nektar::LibUtilities::CommMpi::v_DistGraphCreateAdjacent ( int  indegree,
const int  sources[],
const int  sourceweights[],
int  reorder 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 361 of file CommMpi.cpp.

364{
365#if MPI_VERSION < 3
366 ASSERTL0(false, "MPI_Dist_graph_create_adjacent is not supported in your "
367 "installed MPI version.");
368#else
369 int retval = MPI_Dist_graph_create_adjacent(
370 m_comm, indegree, sources, sourceweights, indegree, sources,
371 sourceweights, MPI_INFO_NULL, reorder, &m_comm);
372
373 ASSERTL0(retval == MPI_SUCCESS,
374 "MPI error performing Dist_graph_create_adjacent.")
375#endif
376}

References ASSERTL0, and m_comm.

◆ v_Finalise()

void Nektar::LibUtilities::CommMpi::v_Finalise ( )
overrideprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 125 of file CommMpi.cpp.

126{
127#ifdef NEKTAR_USING_PETSC
128 PetscFinalize();
129#endif
130 int flag;
131 MPI_Finalized(&flag);
132 if ((!flag) && m_controls_mpi)
133 {
134 MPI_Finalize();
135 }
136}

References m_controls_mpi.

Referenced by Nektar::LibUtilities::CommCwipi::v_Finalise().

◆ v_Gather()

void Nektar::LibUtilities::CommMpi::v_Gather ( void sendbuf,
int  sendcount,
CommDataType  sendtype,
void recvbuf,
int  recvcount,
CommDataType  recvtype,
int  root 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 335 of file CommMpi.cpp.

338{
339 int retval = MPI_Gather(sendbuf, sendcount, sendtype, recvbuf, recvcount,
340 recvtype, root, m_comm);
341
342 ASSERTL0(retval == MPI_SUCCESS, "MPI error performing Gather.");
343}

References ASSERTL0, and m_comm.

◆ v_GetRank()

int Nektar::LibUtilities::CommMpi::v_GetRank ( void  )
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 141 of file CommMpi.cpp.

142{
143 return m_rank;
144}

References m_rank.

◆ v_GetVersion()

std::tuple< int, int, int > Nektar::LibUtilities::CommMpi::v_GetVersion ( )
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 165 of file CommMpi.cpp.

166{
167 int version, subversion;
168 int retval = MPI_Get_version(&version, &subversion);
169
170 ASSERTL0(retval == MPI_SUCCESS, "MPI error performing GetVersion.");
171
172 return std::make_tuple(version, subversion, 0);
173}

References ASSERTL0.

◆ v_Irecv()

void Nektar::LibUtilities::CommMpi::v_Irecv ( void buf,
int  count,
CommDataType  dt,
int  source,
CommRequestSharedPtr  request,
int  loc 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 435 of file CommMpi.cpp.

437{
439 std::static_pointer_cast<CommRequestMpi>(request);
440 MPI_Irecv(buf, count, dt, source, 0, m_comm, req->GetRequest(loc));
441}
std::shared_ptr< CommRequestMpi > CommRequestMpiSharedPtr
Definition: CommMpi.h:86

References CG_Iterations::loc, and m_comm.

◆ v_Irsend()

void Nektar::LibUtilities::CommMpi::v_Irsend ( void buf,
int  count,
CommDataType  dt,
int  dest,
CommRequestSharedPtr  request,
int  loc 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 402 of file CommMpi.cpp.

404{
406 std::static_pointer_cast<CommRequestMpi>(request);
407 MPI_Irsend(buf, count, dt, dest, 0, m_comm, req->GetRequest(loc));
408}

References CG_Iterations::loc, and m_comm.

◆ v_Isend()

void Nektar::LibUtilities::CommMpi::v_Isend ( void buf,
int  count,
CommDataType  dt,
int  dest,
CommRequestSharedPtr  request,
int  loc 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 413 of file CommMpi.cpp.

415{
417 std::static_pointer_cast<CommRequestMpi>(request);
418 MPI_Isend(buf, count, dt, dest, 0, m_comm, req->GetRequest(loc));
419}

References CG_Iterations::loc, and m_comm.

◆ v_IsSerial()

bool Nektar::LibUtilities::CommMpi::v_IsSerial ( void  )
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 157 of file CommMpi.cpp.

158{
159 return m_size == 1;
160}

References Nektar::LibUtilities::Comm::m_size.

◆ v_NeighborAlltoAllv()

void Nektar::LibUtilities::CommMpi::v_NeighborAlltoAllv ( void sendbuf,
int  sendcounts[],
int  sensdispls[],
CommDataType  sendtype,
void recvbuf,
int  recvcounts[],
int  rdispls[],
CommDataType  recvtype 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 381 of file CommMpi.cpp.

386{
387#if MPI_VERSION < 3
388 ASSERTL0(false, "MPI_Neighbor_alltoallv is not supported in your "
389 "installed MPI version.");
390#else
391 int retval =
392 MPI_Neighbor_alltoallv(sendbuf, sendcounts, sdispls, sendtype, recvbuf,
393 recvcounts, rdispls, recvtype, m_comm);
394
395 ASSERTL0(retval == MPI_SUCCESS, "MPI error performing NeighborAllToAllV.");
396#endif
397}

References ASSERTL0, and m_comm.

◆ v_Recv()

void Nektar::LibUtilities::CommMpi::v_Recv ( void buf,
int  count,
CommDataType  dt,
int  source 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 209 of file CommMpi.cpp.

210{
211 MPI_Recv(buf, count, dt, source, 0, m_comm, MPI_STATUS_IGNORE);
212}

References m_comm.

◆ v_RecvInit()

void Nektar::LibUtilities::CommMpi::v_RecvInit ( void buf,
int  count,
CommDataType  dt,
int  source,
CommRequestSharedPtr  request,
int  loc 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 446 of file CommMpi.cpp.

448{
450 std::static_pointer_cast<CommRequestMpi>(request);
451 MPI_Recv_init(buf, count, dt, source, 0, m_comm, req->GetRequest(loc));
452}

References CG_Iterations::loc, and m_comm.

◆ v_Scatter()

void Nektar::LibUtilities::CommMpi::v_Scatter ( void sendbuf,
int  sendcount,
CommDataType  sendtype,
void recvbuf,
int  recvcount,
CommDataType  recvtype,
int  root 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 348 of file CommMpi.cpp.

351{
352 int retval = MPI_Scatter(sendbuf, sendcount, sendtype, recvbuf, recvcount,
353 recvtype, root, m_comm);
354
355 ASSERTL0(retval == MPI_SUCCESS, "MPI error performing Scatter.");
356}

References ASSERTL0, and m_comm.

◆ v_Send()

void Nektar::LibUtilities::CommMpi::v_Send ( void buf,
int  count,
CommDataType  dt,
int  dest 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 194 of file CommMpi.cpp.

195{
196 if (MPISYNC)
197 {
198 MPI_Ssend(buf, count, dt, dest, 0, m_comm);
199 }
200 else
201 {
202 MPI_Send(buf, count, dt, dest, 0, m_comm);
203 }
204}
#define MPISYNC
Definition: CommMpi.h:45

References m_comm, and MPISYNC.

◆ v_SendInit()

void Nektar::LibUtilities::CommMpi::v_SendInit ( void buf,
int  count,
CommDataType  dt,
int  dest,
CommRequestSharedPtr  request,
int  loc 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 424 of file CommMpi.cpp.

426{
428 std::static_pointer_cast<CommRequestMpi>(request);
429 MPI_Send_init(buf, count, dt, dest, 0, m_comm, req->GetRequest(loc));
430}

References CG_Iterations::loc, and m_comm.

◆ v_SendRecv()

void Nektar::LibUtilities::CommMpi::v_SendRecv ( void sendbuf,
int  sendcount,
CommDataType  sendtype,
int  dest,
void recvbuf,
int  recvcount,
CommDataType  recvtype,
int  source 
)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 217 of file CommMpi.cpp.

220{
221 MPI_Status status;
222 int retval = MPI_Sendrecv(sendbuf, sendcount, sendtype, dest, 0, recvbuf,
223 recvcount, recvtype, source, 0, m_comm, &status);
224
225 ASSERTL0(retval == MPI_SUCCESS,
226 "MPI error performing send-receive of data.");
227}

References ASSERTL0, and m_comm.

◆ v_SplitComm()

void Nektar::LibUtilities::CommMpi::v_SplitComm ( int  pRows,
int  pColumns,
int  pTime 
)
overrideprotectedvirtual

Processes are considered as a grid of size pRows*pColumns. Comm objects are created corresponding to the rows and columns of this grid. The row and column to which this process belongs is stored in m_commRow and m_commColumn.

Implements Nektar::LibUtilities::Comm.

Definition at line 495 of file CommMpi.cpp.

496{
497 ASSERTL0(pRows * pColumns * pTime == m_size,
498 "Rows/Columns/Time do not match comm size.");
499
500 MPI_Comm newComm;
501 MPI_Comm gridComm;
502 if (pTime == 1)
503 {
504 // Compute row and column in grid.
505 int myCol = m_rank % pColumns;
506 int myRow = (m_rank - myCol) / pColumns;
507
508 // Split Comm into rows - all processes with same myRow are put in
509 // the same communicator. The rank within this communicator is the
510 // column index.
511 MPI_Comm_split(m_comm, myRow, myCol, &newComm);
512 m_commRow = std::shared_ptr<Comm>(new CommMpi(newComm));
513
514 // Split Comm into columns - all processes with same myCol are put
515 // in the same communicator. The rank within this communicator is
516 // the row index.
517 MPI_Comm_split(m_comm, myCol, myRow, &newComm);
518 m_commColumn = std::shared_ptr<Comm>(new CommMpi(newComm));
519 }
520 else
521 {
522 constexpr int dims = 3;
523 const int sizes[dims] = {pRows, pColumns, pTime};
524 const int periods[dims] = {0, 0, 0};
525 constexpr int reorder = 1;
526
527 MPI_Cart_create(m_comm, dims, sizes, periods, reorder, &gridComm);
528
529 constexpr int keepRow[dims] = {0, 1, 0};
530 MPI_Cart_sub(gridComm, keepRow, &newComm);
531 m_commRow = std::shared_ptr<Comm>(new CommMpi(newComm));
532
533 constexpr int keepCol[dims] = {1, 0, 0};
534 MPI_Cart_sub(gridComm, keepCol, &newComm);
535 m_commColumn = std::shared_ptr<Comm>(new CommMpi(newComm));
536
537 constexpr int keepTime[dims] = {0, 0, 1};
538 MPI_Cart_sub(gridComm, keepTime, &newComm);
539 m_commTime = std::shared_ptr<Comm>(new CommMpi(newComm));
540
541 constexpr int keepSpace[dims] = {1, 1, 0};
542 MPI_Cart_sub(gridComm, keepSpace, &newComm);
543 m_commSpace = std::shared_ptr<Comm>(new CommMpi(newComm));
544 }
545}
CommSharedPtr m_commColumn
Column communicator.
Definition: Comm.h:178
CommSharedPtr m_commRow
Row communicator.
Definition: Comm.h:177
CommSharedPtr m_commTime
Definition: Comm.h:179
CommSharedPtr m_commSpace
Definition: Comm.h:180

References ASSERTL0, CommMpi(), m_comm, Nektar::LibUtilities::Comm::m_commColumn, Nektar::LibUtilities::Comm::m_commRow, Nektar::LibUtilities::Comm::m_commSpace, Nektar::LibUtilities::Comm::m_commTime, m_rank, and Nektar::LibUtilities::Comm::m_size.

◆ v_SplitCommNode()

std::pair< CommSharedPtr, CommSharedPtr > Nektar::LibUtilities::CommMpi::v_SplitCommNode ( )
finalprotectedvirtual

Reimplemented from Nektar::LibUtilities::Comm.

Definition at line 573 of file CommMpi.cpp.

574{
575 std::pair<CommSharedPtr, CommSharedPtr> ret;
576
577#if MPI_VERSION < 3
578 ASSERTL0(false, "Not implemented for non-MPI-3 versions.");
579#else
580 // Create an intra-node communicator.
581 MPI_Comm nodeComm;
582 MPI_Comm_split_type(MPI_COMM_WORLD, MPI_COMM_TYPE_SHARED, m_rank,
583 MPI_INFO_NULL, &nodeComm);
584
585 // For rank 0 of the intra-node communicator, split the main
586 // communicator. Everyone else will get a null communicator.
587 ret.first = std::shared_ptr<Comm>(new CommMpi(nodeComm));
588 ret.second = CommMpi::v_CommCreateIf(ret.first->GetRank() == 0);
589 if (ret.first->GetRank() == 0)
590 {
591 ret.second->SplitComm(1, ret.second->GetSize());
592 }
593#endif
594
595 return ret;
596}
CommSharedPtr v_CommCreateIf(int flag) final
Definition: CommMpi.cpp:550

References ASSERTL0, CommMpi(), m_rank, and v_CommCreateIf().

◆ v_StartAll()

void Nektar::LibUtilities::CommMpi::v_StartAll ( CommRequestSharedPtr  request)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 457 of file CommMpi.cpp.

458{
460 std::static_pointer_cast<CommRequestMpi>(request);
461 if (req->GetNumRequest() != 0)
462 {
463 MPI_Startall(req->GetNumRequest(), req->GetRequest(0));
464 }
465}

◆ v_TreatAsRankZero()

bool Nektar::LibUtilities::CommMpi::v_TreatAsRankZero ( void  )
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 149 of file CommMpi.cpp.

150{
151 return m_rank == 0;
152}

References m_rank.

◆ v_WaitAll()

void Nektar::LibUtilities::CommMpi::v_WaitAll ( CommRequestSharedPtr  request)
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 470 of file CommMpi.cpp.

471{
473 std::static_pointer_cast<CommRequestMpi>(request);
474 if (req->GetNumRequest() != 0)
475 {
476 MPI_Waitall(req->GetNumRequest(), req->GetRequest(0),
477 MPI_STATUSES_IGNORE);
478 }
479}

◆ v_Wtime()

double Nektar::LibUtilities::CommMpi::v_Wtime ( )
finalprotectedvirtual

Implements Nektar::LibUtilities::Comm.

Definition at line 186 of file CommMpi.cpp.

187{
188 return MPI_Wtime();
189}

Member Data Documentation

◆ className

std::string Nektar::LibUtilities::CommMpi::className
static
Initial value:
"ParallelMPI", CommMpi::create, "Parallel communication using MPI.")
static CommSharedPtr create(int narg, char *arg[])
Creates an instance of this class.
Definition: CommMpi.h:93
tKey RegisterCreatorFunction(tKey idKey, CreatorFunction classCreator, std::string pDesc="")
Register a class with the factory.
CommFactory & GetCommFactory()

Name of class.

Definition at line 99 of file CommMpi.h.

◆ m_comm

MPI_Comm Nektar::LibUtilities::CommMpi::m_comm
protected

◆ m_controls_mpi

bool Nektar::LibUtilities::CommMpi::m_controls_mpi
protected

Definition at line 109 of file CommMpi.h.

Referenced by CommMpi(), and v_Finalise().

◆ m_rank

int Nektar::LibUtilities::CommMpi::m_rank {}
protected