Kokkos profiling 2

gyselax · Nov 29, 2023 · e6e6d4c · e6e6d4c
1 parent 1710296
commit e6e6d4c
Show file tree

Hide file tree

Showing 8 changed files with 15 additions and 1 deletion.
diff --git a/src/geometryXVx/poisson/chargedensitycalculator.cpp b/src/geometryXVx/poisson/chargedensitycalculator.cpp
@@ -12,6 +12,7 @@ ChargeDensityCalculator::ChargeDensityCalculator(Quadrature<IDimVx> const& quad)
 
 DSpanX ChargeDensityCalculator::operator()(DSpanX const rho, DViewSpXVx const allfdistribu) const
 {
+    Kokkos::Profiling::pushRegion("ChargeDensityCalculator");
     IndexSp const last_kin_species = allfdistribu.domain<IDimSp>().back();
     IndexSp const last_species = ddc::discrete_space<IDimSp>().charges().domain().back();
     double chargedens_adiabspecies = 0.;
@@ -25,6 +26,7 @@ DSpanX ChargeDensityCalculator::operator()(DSpanX const rho, DViewSpXVx const al
             rho(ix) += charge(isp) * m_quad(allfdistribu[isp][ix]);
         });
     });
+    Kokkos::Profiling::popRegion();
 
     return rho;
 }
diff --git a/src/geometryXVx/poisson/fftpoissonsolver.cpp b/src/geometryXVx/poisson/fftpoissonsolver.cpp
@@ -33,6 +33,7 @@ void FftPoissonSolver::operator()(
         DSpanX const electric_field,
         DViewSpXVx const allfdistribu) const
 {
+    Kokkos::Profiling::pushRegion("PoissonSolver");
     assert(electrostatic_potential.domain() == ddc::get_domain<IDimX>(allfdistribu));
     IDomainX const x_dom = electrostatic_potential.domain();
 
@@ -71,4 +72,5 @@ void FftPoissonSolver::operator()(
 
     // Compute efield = -dPhi/dx where Phi is the electrostatic potential
     m_electric_field(electric_field, electrostatic_potential);
+    Kokkos::Profiling::popRegion();
 }
diff --git a/src/geometryXVx/poisson/splinechargedensitycalculator.cpp b/src/geometryXVx/poisson/splinechargedensitycalculator.cpp
@@ -19,7 +19,7 @@ SplineChargeDensityCalculator::SplineChargeDensityCalculator(
 DSpanX SplineChargeDensityCalculator::operator()(DSpanX const rho, DViewSpXVx const allfdistribu)
         const
 {
-    Kokkos::Profiling::pushRegion("SplineChargeDensityCalculator");
+    Kokkos::Profiling::pushRegion("ChargeDensityCalculator");
     DFieldVx f_vx_slice(allfdistribu.domain<IDimVx>());
     ddc::Chunk<double, BSDomainVx> vx_spline_coef(m_spline_vx_builder.spline_domain());
 

diff --git a/src/geometryXVx/rhs/collisions_inter.cpp b/src/geometryXVx/rhs/collisions_inter.cpp
@@ -94,10 +94,12 @@ device_t<DSpanSpXVx> CollisionsInter::operator()(
         device_t<DSpanSpXVx> allfdistribu_device,
         double dt) const
 {
+    Kokkos::Profiling::pushRegion("CollisionsInter");
     auto allfdistribu_alloc = ddc::create_mirror_view_and_copy(allfdistribu_device);
     ddc::ChunkSpan allfdistribu = allfdistribu_alloc.span_view();
     RK2<DFieldSpXVx> timestepper(allfdistribu.domain());
     timestepper.update(allfdistribu, dt, [&](DSpanSpXVx dy, DViewSpXVx y) { compute_rhs(dy, y); });
     ddc::deepcopy(allfdistribu_device, allfdistribu);
+    Kokkos::Profiling::popRegion();
     return allfdistribu_device;
 }
diff --git a/src/geometryXVx/rhs/collisions_intra.cpp b/src/geometryXVx/rhs/collisions_intra.cpp
@@ -214,6 +214,7 @@ device_t<DSpanSpXVx> CollisionsIntra::operator()(
         device_t<DSpanSpXVx> allfdistribu_device,
         double dt) const
 {
+    Kokkos::Profiling::pushRegion("CollisionsIntra");
     auto allfdistribu_alloc = ddc::create_mirror_view_and_copy(allfdistribu_device);
     ddc::ChunkSpan allfdistribu = allfdistribu_alloc.span_view();
     // density and temperature
@@ -326,5 +327,6 @@ device_t<DSpanSpXVx> CollisionsIntra::operator()(
     });
 
     ddc::deepcopy(allfdistribu_device, allfdistribu);
+    Kokkos::Profiling::popRegion();
     return allfdistribu_device;
 }
diff --git a/src/geometryXVx/rhs/kinetic_source.cpp b/src/geometryXVx/rhs/kinetic_source.cpp
@@ -46,6 +46,7 @@ device_t<DSpanSpXVx> KineticSource::operator()(
         device_t<DSpanSpXVx> const allfdistribu,
         double const dt) const
 {
+    Kokkos::Profiling::pushRegion("KineticSource");
     auto velocity_shape_device_alloc = ddc::create_mirror_view_and_copy(
             Kokkos::DefaultExecutionSpace(),
             m_velocity_shape.span_view());
@@ -68,5 +69,6 @@ device_t<DSpanSpXVx> KineticSource::operator()(
                 allfdistribu(ispxvx) += df;
             });
 
+    Kokkos::Profiling::popRegion();
     return allfdistribu;
 }
diff --git a/src/geometryXVx/rhs/krook_source_adaptive.cpp b/src/geometryXVx/rhs/krook_source_adaptive.cpp
@@ -160,12 +160,14 @@ device_t<DSpanSpXVx> KrookSourceAdaptive::operator()(
         device_t<DSpanSpXVx> const allfdistribu_device,
         double const dt) const
 {
+    Kokkos::Profiling::pushRegion("KrookSource");
     auto allfdistribu_alloc = ddc::create_mirror_view_and_copy(allfdistribu_device);
     ddc::ChunkSpan allfdistribu = allfdistribu_alloc.span_view();
     RK2<DFieldSpXVx> timestepper(allfdistribu.domain());
     timestepper.update(allfdistribu, dt, [&](DSpanSpXVx df, DViewSpXVx f) {
         get_derivative(df, f, allfdistribu);
     });
     ddc::deepcopy(allfdistribu_device, allfdistribu);
+    Kokkos::Profiling::popRegion();
     return allfdistribu_device;
 }
diff --git a/src/geometryXVx/rhs/krook_source_constant.cpp b/src/geometryXVx/rhs/krook_source_constant.cpp
@@ -68,6 +68,7 @@ device_t<DSpanSpXVx> KrookSourceConstant::operator()(
         device_t<DSpanSpXVx> const allfdistribu,
         double const dt) const
 {
+    Kokkos::Profiling::pushRegion("KrookSource");
     auto ftarget_device_alloc = ddc::
             create_mirror_view_and_copy(Kokkos::DefaultExecutionSpace(), m_ftarget.span_view());
     auto ftarget_device = ftarget_device_alloc.span_view();
@@ -90,5 +91,6 @@ device_t<DSpanSpXVx> KrookSourceConstant::operator()(
                                             * dt);
             });
 
+    Kokkos::Profiling::popRegion();
     return allfdistribu;
 }