Profiling region for memory location check

modified: src/synergia/simulation/propagator.cc
fnalacceleratormodeling · Apr 25, 2024 · c0c457b · c0c457b
1 parent 1dc5b79
commit c0c457b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/synergia/simulation/propagator.cc b/src/synergia/simulation/propagator.cc
@@ -49,6 +49,7 @@ Propagator::do_step(Bunch_simulator& simulator,
  // ensure that particles are on the device in case any of the
  // following might have transferred them onto the host
  // custom diagnostics routines, turn_end_action, etc
+ Kokkos::Profiling::pushRegion("memory-location check before step-apply");
  for (auto& train : simulator.get_trains()) {
  for (auto& bunch : train.get_bunches()) {
  auto bparts = bunch.get_bunch_particles();
@@ -58,6 +59,7 @@ Propagator::do_step(Bunch_simulator& simulator,
  }
  }
  }
+ Kokkos::Profiling::popRegion();
 
  // propagate through the step
  step.apply(simulator, logger);