Profile sweeping

fipy/solvers/petsc/petscKrylovSolver.py

@@ -76,6 +76,7 @@ def _adaptPreconditionedTolerance(self, L, x, b):
     def _adaptNaturalTolerance(self, L, x, b):
         return (1., PETSc.KSP.NormType.NATURAL)
 
+    @profile
     def _solve_(self, L, x, b):
         ksp = PETSc.KSP()
         ksp.create(L.comm)

fipy/solvers/pysparse/pysparseSolver.py

@@ -15,6 +15,7 @@ class PysparseSolver(PysparseMatrixSolver):
     .. attention:: This class is abstract. Always create one of its subclasses.
     """
 
+    @profile
     def _solve_(self, L, x, b):
         """
         `_solve_` is only for use by solvers which may use

fipy/solvers/scipy/scipyKrylovSolver.py

@@ -39,6 +39,7 @@ def _adaptInitialTolerance(self, L, x, b):
         factor = self._residualNorm(L, x, b) / self._rhsNorm(L, x, b)
         return (factor, None)
 
+    @profile
     def _solve_(self, L, x, b):
         A = L.matrix
         if self.preconditioner is None:

fipy/solvers/trilinos/trilinosAztecOOSolver.py

@@ -38,6 +38,7 @@ def _adaptInitialTolerance(self, L, x, b):
     def _adaptSolutionTolerance(self, L, x, b):
         return (1., AztecOO.AZ_sol)
 
+    @profile
     def _solve_(self, L, x, b):
 
         Solver = AztecOO.AztecOO(L, x, b)

fipy/terms/term.py

@@ -191,6 +191,7 @@ def solve(self, var=None, solver=None, boundaryConditions=(), dt=None):
 
         self._log.debug("END solve")
 
+    @profile
     def sweep(self, var=None, solver=None, boundaryConditions=(), dt=None, underRelaxation=None, residualFn=None, cacheResidual=False, cacheError=False):
         r"""
         Builds and solves the `Term`'s linear system once. This method