Changeset 17627 for issm/trunk-jpl/src/c/toolkits/mpi/issmmpi.h

Timestamp:

04/02/14 15:36:30 (11 years ago)

Author:

Mathieu Morlighem

Message:

CHG: working on templating MPI calls

File:

• issm/trunk-jpl/src/c/toolkits/mpi/issmmpi.h (modified) (2 diffs)

issm/trunk-jpl/src/c/toolkits/mpi/issmmpi.h

@@ -12 +12 @@
 #error "Cannot compile with HAVE_CONFIG_H symbol! run configure first!"
 #endif
-
 #include <cstddef>
+#include <cassert>
+#include "../../shared/Numerics/types.h"
 
 #ifdef _HAVE_MPI_
@@ -83 +84 @@
         /*}}}*/
 # endif 
+
+template <class T> ISSM_MPI_Datatype TypeToMPIType(){assert(false);};
+template <> inline ISSM_MPI_Datatype TypeToMPIType<IssmDouble>(){return ISSM_MPI_DOUBLE;};
+#if _HAVE_ADOLC_
+template <> inline ISSM_MPI_Datatype TypeToMPIType<IssmPDouble>(){return ISSM_MPI_PDOUBLE;};
+#endif
+template <> inline ISSM_MPI_Datatype TypeToMPIType<int>(){return ISSM_MPI_INT;};
+template <> inline ISSM_MPI_Datatype TypeToMPIType<char>(){return ISSM_MPI_CHAR;};
+template <class T> ISSM_MPI_Datatype ISSM_MPI_Reduce(T *sendbuf,T *recvbuf,int count,ISSM_MPI_Op op, int root, ISSM_MPI_Comm comm){ /*{{{*/
+
+        /*Get MPI type*/
+        ISSM_MPI_Datatype datatype = TypeToMPIType<T>();
+
+        int rc=0;
+#ifdef _HAVE_MPI_
+# ifdef _HAVE_AMPI_
+        rc=AMPI_Reduce(sendbuf,
+                                recvbuf,
+                                count,
+                                datatype,
+                                op,
+                                root,
+                                comm);
+# else
+        rc=MPI_Reduce(sendbuf,
+                                recvbuf,
+                                count,
+                                datatype,
+                                op,
+                                root,
+                                comm);
+# endif
+#else
+# ifdef _HAVE_ADOLC_
+        if (datatype==ISSM_MPI_DOUBLE) { 
+                IssmDouble* activeSendBuf=(IssmDouble*)sendbuf;
+                IssmDouble* activeRecvBuf=(IssmDouble*)recvbuf;
+                for(int i=0;i<count;++i) activeRecvBuf[i]=activeSendBuf[i];
+        }
+        else 
+# endif
+         memcpy(recvbuf,sendbuf,sizeHelper(datatype)*count);
+#endif
+        return rc;
+}/*}}}*/
+template <class T> int ISSM_MPI_Bcast(T *buffer, int count,int root, ISSM_MPI_Comm comm){  /*{{{*/
+
+        int rc=0;
+
+        /*Get MPI type*/
+        ISSM_MPI_Datatype datatype = TypeToMPIType<T>();
+
+#ifdef _HAVE_MPI_
+# ifdef _HAVE_AMPI_
+        rc=AMPI_Bcast(buffer, 
+                                count,
+                                datatype,
+                                root,
+                                comm);
+# else
+        rc=MPI_Bcast(buffer, 
+                                count,
+                                datatype,
+                                root,
+                                comm);
+# endif
+#else 
+        // nothing to be done here 
+#endif
+        return rc;
+}/*}}}*/
+
 /* interfaces  {{{*/
 int ISSM_MPI_Allgather(void *sendbuf, int sendcount, ISSM_MPI_Datatype sendtype, void *recvbuf, int recvcount, ISSM_MPI_Datatype recvtype, ISSM_MPI_Comm comm);