count number of H|psi> operations

electronic-structure · Mar 11, 2020 · 4eb0619 · 4eb0619
1 parent 9e58c5c
commit 4eb0619
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/PW/src/h_psi.f90 b/PW/src/h_psi.f90
@@ -21,6 +21,7 @@ SUBROUTINE h_psi( lda, n, m, psi, hpsi )
   USE mp_bands,           ONLY: use_bgrp_in_hpsi, inter_bgrp_comm
   USE mp,                 ONLY: mp_allgather, mp_size, &
                                 mp_type_create_column_section, mp_type_free
+  USE mp_bands_util,      ONLY: num_loc_op_applied
   !
   IMPLICIT NONE
   !
@@ -42,6 +43,7 @@ SUBROUTINE h_psi( lda, n, m, psi, hpsi )
   INTEGER, ALLOCATABLE :: recv_counts(:), displs(:)
   !
   !
+  num_loc_op_applied = num_loc_op_applied + m
   CALL start_clock( 'h_psi_bgrp' ); !write (*,*) 'start h_psi_bgrp'; FLUSH(6)
   !
   ! band parallelization with non-distributed bands is performed if

diff --git a/PW/src/run_pwscf.f90 b/PW/src/run_pwscf.f90
@@ -58,7 +58,7 @@ SUBROUTINE run_pwscf( exit_status )
   USE mp_world,             ONLY : mpime
   USE dfunct,               ONLY : newd
   USE mod_sirius
-  USE mp_bands_util, ONLY : evp_work_count
+  USE mp_bands_util, ONLY : evp_work_count, num_loc_op_applied
   !
   IMPLICIT NONE
   !
@@ -149,9 +149,11 @@ SUBROUTINE run_pwscf( exit_status )
         CALL electrons()
         IF (use_sirius) THEN
           CALL sirius_get_parameters(sctx, evp_work_count=evp_work_count)
+          CALL sirius_get_parameters(sctx, num_loc_op_applied=num_loc_op_applied)
         ENDIF
         WRITE(stdout, *)
-        WRITE(stdout,'("     evp_work_count : ", F12.6)')evp_work_count
+        WRITE(stdout,'("     evp_work_count     : ", F12.6)')evp_work_count
+        WRITE(stdout,'("     num_loc_op_applied : ", I6)')num_loc_op_applied
         CALL sirius_stop_timer(string("qe|electrons"))
      END IF
      IF (use_sirius.AND.use_sirius_ks_solver) THEN