Add WARPX_PROFILE calls to each python callback. (#2573)

When python callbacks take some time, this is useful as otherwise many
callbacks are lumped together in WarpX::Evolve::step.

1 files changed, 4 insertions, 1 deletions

diff --git a/Source/FieldSolver/ElectrostaticSolver.cpp b/Source/FieldSolver/ElectrostaticSolver.cpp
index f604661bd..5f33bbfb4 100644
--- a/Source/FieldSolver/ElectrostaticSolver.cpp
+++ b/Source/FieldSolver/ElectrostaticSolver.cpp
@@ -172,7 +172,10 @@ WarpX::AddSpaceChargeFieldLabFrame ()
     std::array<Real, 3> beta = {0._rt};
 
     // Compute the potential phi, by solving the Poisson equation
-    if (warpx_py_poissonsolver) warpx_py_poissonsolver();
+    if (warpx_py_poissonsolver) {
+        WARPX_PROFILE("warpx_py_poissonsolver");
+        warpx_py_poissonsolver();
+    }
     else computePhi( rho_fp, phi_fp, beta, self_fields_required_precision,
                      self_fields_absolute_tolerance, self_fields_max_iters,
                      self_fields_verbosity );