some mem purging in MAP lensrec

NextGenCMB · Jul 3, 2023 · d154fa0 · d154fa0
1 parent 063a621
commit d154fa0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/delensalot/core/handler.py b/delensalot/core/handler.py
@@ -906,6 +906,7 @@ def run(self):
                             itlib_iterator.soltn_cond = self.soltn_cond(it)
                             itlib_iterator.iterate(it, 'p')
                             log.info('{}, simidx {} done with it {}'.format(mpi.rank, simidx, it))
+                self.simulationdata.purgecache()
 
             if task == 'calc_meanfield':
                 # TODO I don't like barriers and not sure if they are still needed
@@ -919,6 +920,7 @@ def run(self):
                     self.itlib_iterator = transform(self, iterator_transformer(self, simidx, self.dlensalot_model))
                     for it in range(self.itmax + 1):
                         self.get_blt_it(simidx, it)
+                    self.simulationdata.purgecache()
 
 
     # # @base_exception_handler