From 515edee892e3c0e4c5eda6793ea209ebd5d5e7b0 Mon Sep 17 00:00:00 2001
From: Peter Scherpelz <31747262+peterscherpelz@users.noreply.github.com>
Date: Thu, 18 Nov 2021 16:59:40 -0800
Subject: 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.
---
 Source/Python/WarpXWrappers.cpp | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'Source/Python/WarpXWrappers.cpp')

diff --git a/Source/Python/WarpXWrappers.cpp b/Source/Python/WarpXWrappers.cpp
index 66db34497..6e9b6e941 100644
--- a/Source/Python/WarpXWrappers.cpp
+++ b/Source/Python/WarpXWrappers.cpp
@@ -12,6 +12,7 @@
 #include "Particles/ParticleBoundaryBuffer.H"
 #include "Particles/WarpXParticleContainer.H"
 #include "Utils/WarpXUtil.H"
+#include "Utils/WarpXProfilerWrapper.H"
 #include "WarpX.H"
 #include "WarpXWrappers.H"
 #include "WarpX_py.H"
@@ -166,8 +167,14 @@ extern "C"
     {
         WarpX& warpx = WarpX::GetInstance();
         warpx.InitData();
-        if (warpx_py_afterinit) warpx_py_afterinit();
-        if (warpx_py_particleloader) warpx_py_particleloader();
+        if (warpx_py_afterinit) {
+            WARPX_PROFILE("warpx_py_afterinit");
+            warpx_py_afterinit();
+        }
+        if (warpx_py_particleloader) {
+            WARPX_PROFILE("warpx_py_particleloader");
+            warpx_py_particleloader();
+        }
     }
 
     void warpx_finalize ()
-- 
cgit v1.2.3