Standardize timing ranges

Author: max-Hawkins

src/common/m_mpi_common.fpp

@@ -16,6 +16,8 @@ module m_mpi_common
     use m_derived_types        !< Definitions of the derived types
 
     use m_global_parameters    !< Definitions of the global parameters
+
+    use m_nvtx
     ! ==========================================================================
 
     implicit none
@@ -239,6 +241,7 @@ contains
 #ifdef MFC_SIMULATION
 #ifdef MFC_MPI
 
+        call nvtxStartRange("TSTEP-RUNTIME-INFO-REDUCE-CRIT")
         ! Reducing local extrema of ICFL, VCFL, CCFL and Rc numbers to their
         ! global extrema and bookkeeping the results on the rank 0 processor
         call MPI_REDUCE(icfl_max_loc, icfl_max_glb, 1, &
@@ -253,6 +256,7 @@ contains
                             MPI_DOUBLE_PRECISION, MPI_MIN, 0, &
                             MPI_COMM_WORLD, ierr)
         end if
+        call nvtxEndRange
 
 #else
 

src/simulation/m_mpi_proxy.fpp

@@ -1070,7 +1070,6 @@ contains
                     !$acc update host(q_cons_buff_send, ib_buff_send)
                 #:endif
 
-                call nvtxStartRange("RHS-MPI-SENDRECV")
                 call MPI_SENDRECV( &
                     p_send, buffer_count, MPI_DOUBLE_PRECISION, dst_proc, send_tag, &
                     p_recv, buffer_count, MPI_DOUBLE_PRECISION, src_proc, recv_tag, &

src/simulation/m_rhs.fpp

@@ -669,7 +669,7 @@ contains
             gm_alpha_qp%vf)
         call nvtxEndRange
 
-        call nvtxStartRange("RHS-MPI")
+        call nvtxStartRange("RHS-COMMUNICATION")
         call s_populate_variables_buffers(q_prim_qp%vf, pb, mv)
         call nvtxEndRange
 
@@ -849,6 +849,26 @@ contains
             call nvtxEndRange
             ! END: Additional physics and source terms =========================
 
+            #:if chemistry
+                if (chem_params%advection) then
+                    call nvtxStartRange("RHS-CHEM-ADVECTION")
+
+                    #:for NORM_DIR, XYZ in [(1, 'x'), (2, 'y'), (3, 'z')]
+
+                        if (id == ${NORM_DIR}$) then
+                            call s_compute_chemistry_rhs_${XYZ}$ ( &
+                                flux_n, &
+                                rhs_vf, &
+                                flux_src_n(${NORM_DIR}$)%vf, &
+                                q_prim_vf)
+                        end if
+
+                    #:endfor
+
+                    call nvtxEndRange
+                end if
+            #:endif
+
         end do
         ! END: Dimensional Splitting Loop =================================
 
@@ -883,7 +903,7 @@ contains
         call nvtxEndRange
 
         ! Add bubles source term
-        call nvtxStartRange("RHS_bubbles")
+        call nvtxStartRange("RHS-BUBBLES-SRC")
         if (bubbles .and. (.not. adap_dt) .and. (.not. qbmm)) call s_compute_bubble_source( &
             q_cons_qp%vf(1:sys_size), &
             q_prim_qp%vf(1:sys_size), &