Changeset 23643 for issm/trunk-jpl/src/c/toolkits/petsc/patches/VecToMPISerial.cpp

Timestamp:

01/17/19 15:04:23 (6 years ago)

Author:

Mathieu Morlighem

Message:

CHG: speed up output by not broadcasting to all cpus if only cpu 0 writes the output

File:

• issm/trunk-jpl/src/c/toolkits/petsc/patches/VecToMPISerial.cpp (modified) (3 diffs)

issm/trunk-jpl/src/c/toolkits/petsc/patches/VecToMPISerial.cpp

@@ -12 +12 @@
 #include "../../../shared/shared.h"
 
-int VecToMPISerial(double** pgathered_vector, Vec vector,ISSM_MPI_Comm comm){
+int VecToMPISerial(double** pgathered_vector, Vec vector,ISSM_MPI_Comm comm,bool broadcast){
 
         int i;
@@ -45 +45 @@
 
         /*Allocate gathered vector on all nodes .*/
-        gathered_vector=xNew<double>(vector_size);
+        if(broadcast || my_rank==0){ 
+                gathered_vector=xNew<double>(vector_size);
+        }
 
         /*Allocate local vectors*/
@@ -83 +85 @@
         }
 
-        /*Now, broadcast gathered_vector from node 0 to other nodes: */
-        ISSM_MPI_Bcast(gathered_vector,vector_size,ISSM_MPI_PDOUBLE,0,comm);
+        if(broadcast){
+                /*Now, broadcast gathered_vector from node 0 to other nodes: */
+                ISSM_MPI_Bcast(gathered_vector,vector_size,ISSM_MPI_PDOUBLE,0,comm);
+        }
 
         /*Assign output pointers: */