add profiling

parthenon-hpc-lab · Nov 7, 2023 · 291c6bb · 291c6bb
1 parent 1e42f76
commit 291c6bb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/driver/driver.cpp b/src/driver/driver.cpp
@@ -77,13 +77,16 @@ DriverStatus EvolutionDriver::Execute() {
 
   // Before loop do work
   // App input version
+  Kokkos::Profiling::pushRegion("Driver_UserWorkBeforeLoop");
   if (app_input->UserWorkBeforeLoop != nullptr) {
     app_input->UserWorkBeforeLoop(pmesh, pinput, tm);
   }
+
   // packages version
   for (auto &[name, pkg] : pmesh->packages.AllPackages()) {
     pkg->UserWorkBeforeLoop(pmesh, pinput, tm);
   }
+  Kokkos::Profiling::popRegion(); // Driver_UserWorkBeforeLoop
 
   Kokkos::Profiling::pushRegion("Driver_Main");
   while (tm.KeepGoing()) {