Changeset 25574

Timestamp:

09/15/20 09:05:03 (5 years ago)

Author:

Mathieu Morlighem

Message:

CHG: checkpointing now working in parallelss!

File:

• issm/trunk-jpl/src/c/cores/transient_core.cpp (modified) (2 diffs)

issm/trunk-jpl/src/c/cores/transient_core.cpp

@@ -392 +392 @@
                 femmodel->IceVolumeAboveFloatationx(&J,false);
         }
-        if(IssmComm::GetRank()==0) tape_codi.registerOutput(J);
+        tape_codi.registerOutput(J);
 
         tape_codi.setPassive();
-        J.gradient() = 1.0;
+        if(IssmComm::GetRank()==0) J.gradient() = 1.0;
         tape_codi.evaluate();
 
@@ -465 +465 @@
         }
 
+        /*Broadcast gradient to other ranks (make sure to sum all gradients)*/
+        double* gradient=xNew<double>(Xsize);
+        ISSM_MPI_Allreduce(Xb,gradient,Xsize,ISSM_MPI_PDOUBLE,ISSM_MPI_SUM,IssmComm::GetComm());
+
         /*Now set final gradient*/
         IssmDouble* aG=xNew<IssmDouble>(Xsize);
-        for(int i=0;i<Xsize;i++){
-                aG[i] = reCast<IssmDouble>(Xb[i]);
-        }
+        for(int i=0;i<Xsize;i++) aG[i] = reCast<IssmDouble>(gradient[i]);
+        xDelete<double>(gradient);
         ControlInputSetGradientx(femmodel->elements,femmodel->nodes,femmodel->vertices,femmodel->loads,femmodel->materials,femmodel->parameters,aG);
         xDelete<IssmDouble>(aG);