Skip to content

scalapack/elpa interface: added code to use Nvidia GPUs #384

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
edoapra merged 1 commit into from
Feb 13, 2026
Merged

scalapack/elpa interface: added code to use Nvidia GPUs #384

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 16 additions & 3 deletions global/src/scalapack.F
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3634,6 +3634,7 @@ subroutine ga_evp_real(g_a, g_b, eval, nb8, mout)
logical status
c
character*1 jobz, range, uplo
character*255 env_out
c
integer ha, adra !A
integer hb, adrb !B
Expand Down Expand Up @@ -3685,6 +3686,7 @@ subroutine ga_evp_real(g_a, g_b, eval, nb8, mout)
parameter(one4=1,zero4=0,two4=4,four4=4)
integer info8,dblsize
integer*4 n4
integer err8
c
c processor dependent; machine dependent
c
Expand Down Expand Up @@ -3866,7 +3868,6 @@ subroutine ga_evp_real(g_a, g_b, eval, nb8, mout)
call e%set("nblk", int(nb,kind=c_int), elpa_err) ! size of block-cylic distribution
if(elpa_err.ne.0) call ga_error(
A 'ga-evp_real: e%set local_ncols failed ',elpa_err)

call e%set("mpi_comm_parent", int(elpa_comm,kind=c_int),elpa_err) ! global communicator for all processes which have parts of
if(elpa_err.ne.0) call ga_error(
A 'ga-evp_real: e%set mpi_comm failed ',elpa_err)
Expand Down Expand Up @@ -3900,8 +3901,20 @@ subroutine ga_evp_real(g_a, g_b, eval, nb8, mout)
if(elpa_err.ne.0) call ga_error(
A 'ga-evp_real: get_elpa_row_col failed ',elpa_err)
#endif


#if HAVE_ELPA_NVIDIA
C checks for the following env. variable setting
c ELPA_DEFAULT_real_kernel=ELPA_2STAGE_REAL_NVIDIA_GPU
call get_environment_variable(
A "ELPA_DEFAULT_real_kernel", value=env_out, status=err8)
if(trim(env_out).eq."ELPA_2STAGE_REAL_NVIDIA_GPU") then
call e%set("nvidia-gpu", int(1,c_int), elpa_err)
if(elpa_err.ne.ELPA_OK) call ga_error(
A 'ga-evp_real: e%set nvidia-gpu failed ',elpa_err)
elpa_err = e%setup_gpu()
if(elpa_err.ne.ELPA_OK) call ga_error(
A 'ga-evp_real: e%set setup_gpu failed ',elpa_err)
endif
#endif
#if HAVE_ELPA_2017
#if USE_ALLOCATE
call e%eigenvectors(a, ev, b, elpa_err)
Expand Down
Loading