add a reduction for double&

exanauts · Sep 17, 2018 · d65f2fa · d65f2fa
1 parent 1a42c52
commit d65f2fa
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/examples/plorenz/main.cpp b/examples/plorenz/main.cpp
@@ -125,7 +125,7 @@ int main(int argc, char* argv[]) {
 
     double diff = (H - H_fd).norm();
     diff = diff*diff;
-    MPI_Allreduce(MPI_IN_PLACE, &diff, 1, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
+    paduprop_sum(diff);
     cout << "norm(H - H_fd)" << endl << sqrt(diff) << endl;
     if(diff > 1e-4) {
       cerr << "ERROR: HC and AD Hessian differ." << endl;
@@ -151,7 +151,7 @@ int main(int argc, char* argv[]) {
     drivers.fdT_driver(x, T_fd, start, end);
     double diff = (T - T_fd).norm();
     diff = diff*diff;
-    MPI_Allreduce(MPI_IN_PLACE, &diff, 1, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
+    paduprop_sum(diff);
     cout << "norm(T - T_fd)" << endl << sqrt(diff) << endl;
     if(diff > 1e-5) {
       cerr << "ERROR: HC and AD tensor differ." << endl;

diff --git a/include/parallel.hpp b/include/parallel.hpp
@@ -15,6 +15,7 @@ size_t paduprop_getend(size_t dim);
 
 int paduprop_getrank();
 int paduprop_getcommsize();
+int paduprop_sum(double&);
 int paduprop_sum(pMatrix<double>&);
 int paduprop_sum(pVector<double> &vec);
 int paduprop_gather(pTensor3<double> &ten);

diff --git a/src/parallel.cpp b/src/parallel.cpp
@@ -91,6 +91,13 @@ int paduprop_getcommsize() {
 #endif
   return size;
 }
+int paduprop_sum(double &v) {
+  int ierr = 0;
+#ifdef MPI_VERSION
+  ierr = MPI_Allreduce(MPI_IN_PLACE, &v, 1, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
+#endif
+  return ierr;
+}
 int paduprop_sum(pMatrix<double> &mat) {
   size_t nrows = mat.nrows();
   size_t ncols = mat.ncols();