mirror of
https://github.com/NGSolve/netgen.git
synced 2024-12-27 22:30:33 +05:00
360 lines
8.9 KiB
C++
360 lines
8.9 KiB
C++
#ifndef FILE_PARALLEL
|
|
#define FILE_PARALLEL
|
|
|
|
|
|
|
|
#ifdef VTRACE
|
|
#include "vt_user.h"
|
|
#else
|
|
#define VT_USER_START(n)
|
|
#define VT_USER_END(n)
|
|
#define VT_TRACER(n)
|
|
#endif
|
|
|
|
|
|
namespace netgen
|
|
{
|
|
|
|
extern DLL_HEADER int id, ntasks;
|
|
|
|
|
|
#ifdef PARALLEL
|
|
|
|
enum { MPI_TAG_CMD = 110 };
|
|
enum { MPI_TAG_MESH = 210 };
|
|
enum { MPI_TAG_VIS = 310 };
|
|
|
|
extern MPI_Comm mesh_comm;
|
|
|
|
template <class T>
|
|
MPI_Datatype MyGetMPIType ( )
|
|
{ cerr << "ERROR in GetMPIType() -- no type found" << endl;return 0; }
|
|
|
|
template <>
|
|
inline MPI_Datatype MyGetMPIType<int> ( )
|
|
{ return MPI_INT; }
|
|
|
|
template <>
|
|
inline MPI_Datatype MyGetMPIType<double> ( )
|
|
{ return MPI_DOUBLE; }
|
|
|
|
|
|
template <int S> class Vec;
|
|
template <>
|
|
inline MPI_Datatype MyGetMPIType<Vec<3> > ()
|
|
{
|
|
static MPI_Datatype MPI_T = 0;
|
|
if (!MPI_T)
|
|
{
|
|
MPI_Type_contiguous ( 3, MPI_DOUBLE, &MPI_T);
|
|
MPI_Type_commit ( &MPI_T );
|
|
}
|
|
return MPI_T;
|
|
};
|
|
|
|
|
|
inline void MyMPI_Send (int i, int dest, int tag)
|
|
{
|
|
int hi = i;
|
|
MPI_Send( &hi, 1, MPI_INT, dest, tag, MPI_COMM_WORLD);
|
|
}
|
|
|
|
inline void MyMPI_Recv (int & i, int src, int tag)
|
|
{
|
|
MPI_Status status;
|
|
MPI_Recv( &i, 1, MPI_INT, src, tag, MPI_COMM_WORLD, &status);
|
|
}
|
|
|
|
|
|
|
|
inline void MyMPI_Send (const string & s, int dest, int tag)
|
|
{
|
|
MPI_Send( const_cast<char*> (s.c_str()), s.length(), MPI_CHAR, dest, tag, MPI_COMM_WORLD);
|
|
}
|
|
|
|
inline void MyMPI_Recv (string & s, int src, int tag)
|
|
{
|
|
MPI_Status status;
|
|
int len;
|
|
MPI_Probe (src, tag, MPI_COMM_WORLD, &status);
|
|
MPI_Get_count (&status, MPI_CHAR, &len);
|
|
s.assign (len, ' ');
|
|
MPI_Recv( &s[0], len, MPI_CHAR, src, tag, MPI_COMM_WORLD, &status);
|
|
}
|
|
|
|
|
|
|
|
|
|
template <class T, int BASE>
|
|
inline void MyMPI_Send (FlatArray<T, BASE> s, int dest, int tag)
|
|
{
|
|
MPI_Send( &s.First(), s.Size(), MyGetMPIType<T>(), dest, tag, MPI_COMM_WORLD);
|
|
}
|
|
|
|
template <class T, int BASE>
|
|
inline void MyMPI_Recv ( FlatArray<T, BASE> s, int src, int tag)
|
|
{
|
|
MPI_Status status;
|
|
MPI_Recv( &s.First(), s.Size(), MyGetMPIType<T>(), src, tag, MPI_COMM_WORLD, &status);
|
|
}
|
|
|
|
template <class T, int BASE>
|
|
inline void MyMPI_Recv ( Array <T, BASE> & s, int src, int tag)
|
|
{
|
|
MPI_Status status;
|
|
int len;
|
|
MPI_Probe (src, tag, MPI_COMM_WORLD, &status);
|
|
MPI_Get_count (&status, MyGetMPIType<T>(), &len);
|
|
|
|
s.SetSize (len);
|
|
MPI_Recv( &s.First(), len, MyGetMPIType<T>(), src, tag, MPI_COMM_WORLD, &status);
|
|
}
|
|
|
|
template <class T, int BASE>
|
|
inline int MyMPI_Recv ( Array <T, BASE> & s, int tag)
|
|
{
|
|
MPI_Status status;
|
|
int len;
|
|
MPI_Probe (MPI_ANY_SOURCE, tag, MPI_COMM_WORLD, &status);
|
|
|
|
int src = status.MPI_SOURCE;
|
|
|
|
MPI_Get_count (&status, MyGetMPIType<T>(), &len);
|
|
|
|
s.SetSize (len);
|
|
MPI_Recv( &s.First(), len, MyGetMPIType<T>(), src, tag, MPI_COMM_WORLD, &status);
|
|
|
|
return src;
|
|
}
|
|
|
|
|
|
/*
|
|
template <class T, int BASE>
|
|
inline void MyMPI_ISend (FlatArray<T, BASE> s, int dest, int tag, MPI_Request & request)
|
|
{
|
|
MPI_Isend( &s.First(), s.Size(), MyGetMPIType<T>(), dest, tag, MPI_COMM_WORLD, & request);
|
|
}
|
|
|
|
|
|
template <class T, int BASE>
|
|
inline void MyMPI_IRecv (FlatArray<T, BASE> s, int dest, int tag, MPI_Request & request)
|
|
{
|
|
MPI_Irecv( &s.First(), s.Size(), MyGetMPIType<T>(), dest, tag, MPI_COMM_WORLD, & request);
|
|
}
|
|
*/
|
|
|
|
template <class T, int BASE>
|
|
inline MPI_Request MyMPI_ISend (FlatArray<T, BASE> s, int dest, int tag, MPI_Comm comm = MPI_COMM_WORLD)
|
|
{
|
|
MPI_Request request;
|
|
MPI_Isend( &s.First(), s.Size(), MyGetMPIType<T>(), dest, tag, comm, &request);
|
|
return request;
|
|
}
|
|
|
|
|
|
template <class T, int BASE>
|
|
inline MPI_Request MyMPI_IRecv (FlatArray<T, BASE> s, int dest, int tag, MPI_Comm comm = MPI_COMM_WORLD)
|
|
{
|
|
MPI_Request request;
|
|
MPI_Irecv( &s.First(), s.Size(), MyGetMPIType<T>(), dest, tag, comm, &request);
|
|
return request;
|
|
}
|
|
|
|
/*
|
|
template <class T, int BASE>
|
|
inline void MyMPI_ISend (FlatArray<T, BASE> s, int dest, int tag)
|
|
{
|
|
MPI_Request request;
|
|
MPI_Isend( &s.First(), s.Size(), MyGetMPIType<T>(), dest, tag, MPI_COMM_WORLD, &request);
|
|
MPI_Request_free (&request);
|
|
}
|
|
|
|
|
|
template <class T, int BASE>
|
|
inline void MyMPI_IRecv (FlatArray<T, BASE> s, int dest, int tag)
|
|
{
|
|
MPI_Request request;
|
|
MPI_Irecv( &s.First(), s.Size(), MyGetMPIType<T>(), dest, tag, MPI_COMM_WORLD, &request);
|
|
MPI_Request_free (&request);
|
|
}
|
|
*/
|
|
|
|
|
|
|
|
/*
|
|
send a table entry to each of the prcesses in the group ...
|
|
receive-table entries will be set
|
|
*/
|
|
|
|
/*
|
|
template <typename T>
|
|
inline void MyMPI_ExchangeTable (TABLE<T> & send_data,
|
|
TABLE<T> & recv_data, int tag,
|
|
MPI_Comm comm = MPI_COMM_WORLD)
|
|
{
|
|
int ntasks, rank;
|
|
MPI_Comm_size(comm, &ntasks);
|
|
MPI_Comm_rank(comm, &rank);
|
|
|
|
Array<MPI_Request> requests;
|
|
for (int dest = 0; dest < ntasks; dest++)
|
|
if (dest != rank)
|
|
requests.Append (MyMPI_ISend (send_data[dest], dest, tag, comm));
|
|
|
|
for (int i = 0; i < ntasks-1; i++)
|
|
{
|
|
MPI_Status status;
|
|
MPI_Probe (MPI_ANY_SOURCE, tag, comm, &status);
|
|
int size, src = status.MPI_SOURCE;
|
|
MPI_Get_count (&status, MPI_INT, &size);
|
|
recv_data.SetEntrySize (src, size, sizeof(T));
|
|
requests.Append (MyMPI_IRecv (recv_data[src], src, tag, comm));
|
|
}
|
|
MPI_Barrier (comm);
|
|
MPI_Waitall (requests.Size(), &requests[0], MPI_STATUS_IGNORE);
|
|
}
|
|
*/
|
|
|
|
template <typename T>
|
|
inline void MyMPI_ExchangeTable (TABLE<T> & send_data,
|
|
TABLE<T> & recv_data, int tag,
|
|
MPI_Comm comm = MPI_COMM_WORLD)
|
|
{
|
|
int ntasks, rank;
|
|
MPI_Comm_size(comm, &ntasks);
|
|
MPI_Comm_rank(comm, &rank);
|
|
|
|
Array<int> send_sizes(ntasks);
|
|
Array<int> recv_sizes(ntasks);
|
|
for (int i = 0; i < ntasks; i++)
|
|
send_sizes[i] = send_data[i].Size();
|
|
|
|
MPI_Alltoall (&send_sizes[0], 1, MPI_INT,
|
|
&recv_sizes[0], 1, MPI_INT, comm);
|
|
|
|
// in-place is buggy !
|
|
// MPI_Alltoall (MPI_IN_PLACE, 1, MPI_INT,
|
|
// &recv_sizes[0], 1, MPI_INT, comm);
|
|
|
|
|
|
for (int i = 0; i < ntasks; i++)
|
|
recv_data.SetEntrySize (i, recv_sizes[i], sizeof(T));
|
|
|
|
Array<MPI_Request> requests;
|
|
for (int dest = 0; dest < ntasks; dest++)
|
|
if (dest != rank && send_data[dest].Size())
|
|
requests.Append (MyMPI_ISend (send_data[dest], dest, tag, comm));
|
|
|
|
for (int dest = 0; dest < ntasks; dest++)
|
|
if (dest != rank && recv_data[dest].Size())
|
|
requests.Append (MyMPI_IRecv (recv_data[dest], dest, tag, comm));
|
|
|
|
// MPI_Barrier (comm);
|
|
MPI_Waitall (requests.Size(), &requests[0], MPI_STATUS_IGNORE);
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
extern void MyMPI_SendCmd (const char * cmd);
|
|
extern string MyMPI_RecvCmd ();
|
|
|
|
|
|
|
|
|
|
template <class T>
|
|
inline void MyMPI_Bcast (T & s, MPI_Comm comm = MPI_COMM_WORLD)
|
|
{
|
|
MPI_Bcast (&s, 1, MyGetMPIType<T>(), 0, comm);
|
|
}
|
|
|
|
template <class T>
|
|
inline void MyMPI_Bcast (Array<T, 0> & s, MPI_Comm comm = MPI_COMM_WORLD)
|
|
{
|
|
int size = s.Size();
|
|
MyMPI_Bcast (size, comm);
|
|
if (id != 0) s.SetSize (size);
|
|
MPI_Bcast (&s[0], size, MyGetMPIType<T>(), 0, comm);
|
|
}
|
|
|
|
template <class T>
|
|
inline void MyMPI_Bcast (Array<T, 0> & s, int root, MPI_Comm comm = MPI_COMM_WORLD)
|
|
{
|
|
int id;
|
|
MPI_Comm_rank(comm, &id);
|
|
|
|
int size = s.Size();
|
|
MPI_Bcast (&size, 1, MPI_INT, root, comm);
|
|
if (id != root) s.SetSize (size);
|
|
if ( !size ) return;
|
|
MPI_Bcast (&s[0], size, MyGetMPIType<T>(), root, comm);
|
|
}
|
|
|
|
template <class T, class T2>
|
|
inline void MyMPI_Allgather (const T & send, FlatArray<T2> recv, MPI_Comm comm)
|
|
{
|
|
MPI_Allgather( const_cast<T*> (&send), 1, MyGetMPIType<T>(), &recv[0], 1, MyGetMPIType<T2>(), comm);
|
|
}
|
|
|
|
template <class T, class T2>
|
|
inline void MyMPI_Alltoall (FlatArray<T> send, FlatArray<T2> recv, MPI_Comm comm)
|
|
{
|
|
MPI_Alltoall( &send[0], 1, MyGetMPIType<T>(), &recv[0], 1, MyGetMPIType<T2>(), comm);
|
|
}
|
|
|
|
// template <class T, class T2>
|
|
// inline void MyMPI_Alltoall_Block (FlatArray<T> send, FlatArray<T2> recv, int blocklen, MPI_Comm comm)
|
|
// {
|
|
// MPI_Alltoall( &send[0], blocklen, MyGetMPIType<T>(), &recv[0], blocklen, MyGetMPIType<T2>(), comm);
|
|
// }
|
|
|
|
|
|
|
|
/*
|
|
inline void MyMPI_Send ( int *& s, int len, int dest, int tag)
|
|
{
|
|
int hlen = len;
|
|
MPI_Send( &hlen, 1, MPI_INT, dest, tag, MPI_COMM_WORLD);
|
|
MPI_Send( s, len, MPI_INT, dest, tag, MPI_COMM_WORLD);
|
|
}
|
|
|
|
|
|
inline void MyMPI_Recv ( int *& s, int & len, int src, int tag)
|
|
{
|
|
MPI_Status status;
|
|
MPI_Recv( &len, 1, MPI_INT, src, tag, MPI_COMM_WORLD, &status);
|
|
if ( s )
|
|
delete [] s;
|
|
s = new int [len];
|
|
MPI_Recv( s, len, MPI_INT, src, tag, MPI_COMM_WORLD, &status);
|
|
}
|
|
|
|
|
|
|
|
inline void MyMPI_Send ( double * s, int len, int dest, int tag)
|
|
{
|
|
MPI_Send( &len, 1, MPI_INT, dest, tag, MPI_COMM_WORLD);
|
|
MPI_Send( s, len, MPI_DOUBLE, dest, tag, MPI_COMM_WORLD);
|
|
}
|
|
|
|
|
|
inline void MyMPI_Recv ( double *& s, int & len, int src, int tag)
|
|
{
|
|
MPI_Status status;
|
|
MPI_Recv( &len, 1, MPI_INT, src, tag, MPI_COMM_WORLD, &status);
|
|
if ( s )
|
|
delete [] s;
|
|
s = new double [len];
|
|
MPI_Recv( s, len, MPI_DOUBLE, src, tag, MPI_COMM_WORLD, &status);
|
|
}
|
|
*/
|
|
|
|
#endif // PARALLEL
|
|
|
|
}
|
|
|
|
#endif
|