Added BL_PROFILE statements to implicit solvers (BLAST-WarpX#5783)

debog · web-flow · commit f3393f1dfdda · 2025-03-22T01:04:56.000Z
diff --git a/Source/FieldSolver/ImplicitSolvers/SemiImplicitEM.cpp b/Source/FieldSolver/ImplicitSolvers/SemiImplicitEM.cpp
@@ -13,6 +13,8 @@ using namespace amrex::literals;
 
 void SemiImplicitEM::Define ( WarpX*  a_WarpX )
 {
+    BL_PROFILE("SemiImplicitEM::Define()");
+
     WARPX_ALWAYS_ASSERT_WITH_MESSAGE(
         !m_is_defined,
         "SemiImplicitEM object is already defined!");
@@ -57,6 +59,8 @@ void SemiImplicitEM::OneStep ( amrex::Real  start_time,
                                amrex::Real  a_dt,
                                int          a_step )
 {
+    BL_PROFILE("SemiImplicitEM::OneStep()");
+
     amrex::ignore_unused(a_step);
 
     // Set the member time step
@@ -107,6 +111,8 @@ void SemiImplicitEM::ComputeRHS ( WarpXSolverVec&  a_RHS,
                                   int              a_nl_iter,
                                   bool             a_from_jacobian )
 {
+    BL_PROFILE("SemiImplicitEM::ComputeRHS()");
+
     // Update WarpX-owned Efield_fp using current state of Eg from
     // the nonlinear solver at time n+theta
     const amrex::Real half_time = start_time + 0.5_rt*m_dt;
diff --git a/Source/FieldSolver/ImplicitSolvers/ThetaImplicitEM.cpp b/Source/FieldSolver/ImplicitSolvers/ThetaImplicitEM.cpp
@@ -14,6 +14,8 @@ using namespace amrex::literals;
 
 void ThetaImplicitEM::Define ( WarpX* const  a_WarpX )
 {
+    BL_PROFILE("ThetaImplicitEM::Define()");
+
     WARPX_ALWAYS_ASSERT_WITH_MESSAGE(
         !m_is_defined,
         "ThetaImplicitEM object is already defined!");
@@ -77,6 +79,8 @@ void ThetaImplicitEM::Define ( WarpX* const  a_WarpX )
 
 void ThetaImplicitEM::PrintParameters () const
 {
+    BL_PROFILE("ThetaImplicitEM::PrintParameters()");
+
     if (!m_WarpX->Verbose()) { return; }
     amrex::Print() << "\n";
     amrex::Print() << "-----------------------------------------------------------\n";
@@ -99,6 +103,8 @@ void ThetaImplicitEM::OneStep ( const amrex::Real  start_time,
                                 const amrex::Real  a_dt,
                                 const int          a_step )
 {
+    BL_PROFILE("ThetaImplicitEM::OneStep()");
+
     amrex::ignore_unused(a_step);
 
     // Fields have Eg^{n} and Bg^{n}
@@ -147,6 +153,8 @@ void ThetaImplicitEM::ComputeRHS ( WarpXSolverVec&  a_RHS,
                                    int              a_nl_iter,
                                    bool             a_from_jacobian )
 {
+    BL_PROFILE("ThetaImplicitEM::ComputeRHS()");
+
     // Update WarpX-owned Efield_fp and Bfield_fp using current state of
     // Eg from the nonlinear solver at time n+theta
     UpdateWarpXFields( a_E, start_time );
@@ -163,6 +171,7 @@ void ThetaImplicitEM::ComputeRHS ( WarpXSolverVec&  a_RHS,
 void ThetaImplicitEM::UpdateWarpXFields ( const WarpXSolverVec&  a_E,
                                           amrex::Real start_time )
 {
+    BL_PROFILE("ThetaImplicitEM::UpdateWarpXFields()");
 
     // Update Efield_fp owned by WarpX
     const amrex::Real theta_time = start_time + m_theta*m_dt;
@@ -176,6 +185,7 @@ void ThetaImplicitEM::UpdateWarpXFields ( const WarpXSolverVec&  a_E,
 
 void ThetaImplicitEM::FinishFieldUpdate ( amrex::Real end_time )
 {
+    BL_PROFILE("ThetaImplicitEM::FinishFieldUpdate()");
 
     // Eg^{n+1} = (1/theta)*Eg^{n+theta} + (1-1/theta)*Eg^n
     // Bg^{n+1} = (1/theta)*Bg^{n+theta} + (1-1/theta)*Bg^n
diff --git a/Source/NonlinearSolvers/CurlCurlMLMGPC.H b/Source/NonlinearSolvers/CurlCurlMLMGPC.H
@@ -192,6 +192,7 @@ template <class T, class Ops>
 void CurlCurlMLMGPC<T,Ops>::Define ( const T& a_U,
                                      Ops* const a_ops )
 {
+    BL_PROFILE("CurlCurlMLMGPC::Define()");
     using namespace amrex;
 
     WARPX_ALWAYS_ASSERT_WITH_MESSAGE(
@@ -265,6 +266,7 @@ void CurlCurlMLMGPC<T,Ops>::Define ( const T& a_U,
 template <class T, class Ops>
 void CurlCurlMLMGPC<T,Ops>::Update (const T& a_U)
 {
+    BL_PROFILE("CurlCurlMLMGPC::Update()");
     using namespace amrex;
 
     WARPX_ALWAYS_ASSERT_WITH_MESSAGE(
@@ -316,6 +318,7 @@ void CurlCurlMLMGPC<T,Ops>::Apply (T& a_x, const T& a_b)
     //  operator:
     //      A x = curl (alpha * curl (x) ) + beta * x
 
+    BL_PROFILE("CurlCurlMLMGPC::Apply()");
     using namespace amrex;
 
     WARPX_ALWAYS_ASSERT_WITH_MESSAGE(
diff --git a/Source/NonlinearSolvers/JacobiPC.H b/Source/NonlinearSolvers/JacobiPC.H
@@ -151,6 +151,7 @@ template <class T, class Ops>
 void JacobiPC<T,Ops>::Define ( const T& a_U,
                                Ops* const a_ops )
 {
+    BL_PROFILE("JacobiPC::Define()");
     using namespace amrex;
 
     WARPX_ALWAYS_ASSERT_WITH_MESSAGE(
@@ -177,6 +178,7 @@ void JacobiPC<T,Ops>::Define ( const T& a_U,
 template <class T, class Ops>
 void JacobiPC<T,Ops>::Update (const T& a_U)
 {
+    BL_PROFILE("JacobiPC::Update()");
     using namespace amrex;
 
     WARPX_ALWAYS_ASSERT_WITH_MESSAGE(
@@ -199,6 +201,7 @@ void JacobiPC<T,Ops>::Apply (T& a_x, const T& a_b)
     //      A x = b
     //  where A is the linear operator.
 
+    BL_PROFILE("JacobiPC::Apply()");
     using namespace amrex;
 
     WARPX_ALWAYS_ASSERT_WITH_MESSAGE(
diff --git a/Source/NonlinearSolvers/JacobianFunctionMF.H b/Source/NonlinearSolvers/JacobianFunctionMF.H
@@ -182,6 +182,7 @@ void JacobianFunctionMF<T,Ops>::define ( const T& a_U,
                                          Ops* a_ops,
                                          const PreconditionerType& a_pc_type )
 {
+    BL_PROFILE("JacobianFunctionMF::::define()");
     m_Z.Define(a_U);
     m_Y0.Define(a_U);
     m_R0.Define(a_U);
@@ -211,6 +212,7 @@ void JacobianFunctionMF<T,Ops>::define ( const T& a_U,
 template <class T, class Ops>
 auto JacobianFunctionMF<T,Ops>::makeVecRHS () const -> T
 {
+    BL_PROFILE("JacobianFunctionMF::::makeVecRHS()");
     T Vec;
     Vec.Define(m_R);
     return Vec;
@@ -219,6 +221,7 @@ auto JacobianFunctionMF<T,Ops>::makeVecRHS () const -> T
 template <class T, class Ops>
 auto JacobianFunctionMF<T,Ops>::makeVecLHS () const -> T
 {
+    BL_PROFILE("JacobianFunctionMF::::makeVecLHS()");
     T Vec;
     Vec.Define(m_R);
     return Vec;
diff --git a/Source/NonlinearSolvers/NewtonSolver.H b/Source/NonlinearSolvers/NewtonSolver.H
@@ -180,6 +180,7 @@ template <class Vec, class Ops>
 void NewtonSolver<Vec,Ops>::Define ( const Vec&  a_U,
                                      Ops*        a_ops )
 {
+    BL_PROFILE("NewtonSolver::Define()");
     WARPX_ALWAYS_ASSERT_WITH_MESSAGE(
         !this->m_is_defined,
         "Newton nonlinear solver object is already defined!");
@@ -389,7 +390,7 @@ void NewtonSolver<Vec,Ops>::EvalResidual ( Vec&         a_F,
                                            amrex::Real  a_time,
                                            int          a_iter ) const
 {
-
+    BL_PROFILE("NewtonSolver::EvalResidual()");
     m_ops->ComputeRHS( m_R, a_U, a_time, a_iter, false );
 
     // set base U and R(U) for matrix-free Jacobian action calculation
