Commit 6ffc22ac authored by Uliana Alekseeva's avatar Uliana Alekseeva

nvtx stuff added

parent ba479238
...@@ -28,6 +28,9 @@ CONTAINS ...@@ -28,6 +28,9 @@ CONTAINS
USE m_types USE m_types
#ifdef CPP_ELPA_ONENODE #ifdef CPP_ELPA_ONENODE
USE elpa USE elpa
#endif
#ifdef CPP_GPU
USE nvtx
#endif #endif
IMPLICIT NONE IMPLICIT NONE
...@@ -45,7 +48,9 @@ CONTAINS ...@@ -45,7 +48,9 @@ CONTAINS
INTEGER :: kernel INTEGER :: kernel
CLASS(elpa_t),pointer :: elpa_obj CLASS(elpa_t),pointer :: elpa_obj
print*, "ELPA 20180525 started" #ifdef CPP_GPU
call nvtxStartRange("ELPA",5)
#endif
err = elpa_init(20180525) err = elpa_init(20180525)
elpa_obj => elpa_allocate() elpa_obj => elpa_allocate()
...@@ -65,15 +70,23 @@ CONTAINS ...@@ -65,15 +70,23 @@ CONTAINS
CALL elpa_obj%set("gpu",1,err) CALL elpa_obj%set("gpu",1,err)
#endif #endif
err = elpa_obj%setup() err = elpa_obj%setup()
call elpa_obj%get("solver", kernel)
print *, "elpa uses " // elpa_int_value_to_string("solver", kernel) // " solver"
CALL hmat%add_transpose(hmat) CALL hmat%add_transpose(hmat)
CALL smat%add_transpose(smat) CALL smat%add_transpose(smat)
#ifdef CPP_GPU
call nvtxStartRange("EigVec",7)
#endif
IF (hmat%l_real) THEN IF (hmat%l_real) THEN
CALL elpa_obj%generalized_eigenvectors(hmat%data_r,smat%data_r,eig2, ev_dist%data_r, .FALSE.,err) CALL elpa_obj%generalized_eigenvectors(hmat%data_r,smat%data_r,eig2, ev_dist%data_r, .FALSE.,err)
ELSE ELSE
CALL elpa_obj%generalized_eigenvectors(hmat%data_c,smat%data_c,eig2, ev_dist%data_c, .FALSE., err) CALL elpa_obj%generalized_eigenvectors(hmat%data_c,smat%data_c,eig2, ev_dist%data_c, .FALSE., err)
ENDIF ENDIF
#ifdef CPP_GPU
call nvtxEndRange!("EigVec",8)
#endif
CALL elpa_deallocate(elpa_obj) CALL elpa_deallocate(elpa_obj)
CALL elpa_uninit() CALL elpa_uninit()
...@@ -86,6 +99,9 @@ CONTAINS ...@@ -86,6 +99,9 @@ CONTAINS
CALL ev%alloc(hmat%l_real,hmat%matsize1,ne) CALL ev%alloc(hmat%l_real,hmat%matsize1,ne)
CALL ev%copy(ev_dist,1,1) CALL ev%copy(ev_dist,1,1)
#ifdef CPP_GPU
call nvtxEndRange!("ELPA",7)
#endif
#endif #endif
END SUBROUTINE elpa_diag_onenode END SUBROUTINE elpa_diag_onenode
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment