mirror of https://gitlab.com/QEF/q-e.git
166 lines
4.1 KiB
Fortran
Executable File
166 lines
4.1 KiB
Fortran
Executable File
!
|
|
! Copyright (C) 2001-2005 Quantum-ESPRESSO group
|
|
! This file is distributed under the terms of the
|
|
! GNU General Public License. See the file `License'
|
|
! in the root directory of the present distribution,
|
|
! or http://www.gnu.org/copyleft/gpl.txt .
|
|
!
|
|
#include "f_defs.h"
|
|
!
|
|
!----------------------------------------------------------------------------
|
|
SUBROUTINE rdiagh( n, h, ldh, e, v )
|
|
!----------------------------------------------------------------------------
|
|
!
|
|
! ... calculates all the eigenvalues and eigenvectors of a real
|
|
! ... simmetric matrix H . On output, the matrix is unchanged
|
|
!
|
|
USE kinds, ONLY : DP
|
|
USE control_flags, ONLY : use_para_diago, para_diago_dim
|
|
USE mp_global, ONLY : nproc, npool, nproc_pool, me_pool, &
|
|
root_pool, intra_pool_comm, my_image_id
|
|
USE mp, ONLY : mp_bcast
|
|
USE parallel_toolkit, ONLY : diagonalize
|
|
!
|
|
IMPLICIT NONE
|
|
!
|
|
! ... on INPUT
|
|
!
|
|
INTEGER :: n, ldh
|
|
! dimension of the matrix to be diagonalized
|
|
! leading dimension of h, as declared in the calling pgm unit
|
|
REAL(DP) :: h(ldh,n)
|
|
! matrix to be diagonalized
|
|
!
|
|
! ... on OUTPUT
|
|
!
|
|
REAL(DP) :: e(n) ! eigenvalues
|
|
REAL(DP) :: v(ldh,n) ! eigenvectors (column-wise)
|
|
!
|
|
!
|
|
CALL start_clock( 'rdiagh' )
|
|
!
|
|
IF ( use_para_diago .AND. n > para_diago_dim ) THEN
|
|
!
|
|
CALL diagonalize( 1, h, ldh, e, v, ldh, n, &
|
|
nproc_pool, me_pool, intra_pool_comm )
|
|
!
|
|
ELSE
|
|
!
|
|
#if defined (__AIX)
|
|
CALL rdiagh_aix()
|
|
#else
|
|
CALL rdiagh_lapack()
|
|
#endif
|
|
!
|
|
END IF
|
|
!
|
|
CALL stop_clock( 'rdiagh' )
|
|
!
|
|
RETURN
|
|
!
|
|
CONTAINS
|
|
!
|
|
! ... internal procedures
|
|
!
|
|
#if defined (__AIX)
|
|
!
|
|
!-----------------------------------------------------------------------
|
|
SUBROUTINE rdiagh_aix()
|
|
!-----------------------------------------------------------------------
|
|
!
|
|
IMPLICIT NONE
|
|
!
|
|
! ... local variables (ESSL version)
|
|
!
|
|
INTEGER :: naux, i, j, ij
|
|
COMPLEX(DP), ALLOCATABLE :: hp(:), aux(:)
|
|
!
|
|
!
|
|
naux = 4 * n
|
|
!
|
|
ALLOCATE( hp( n * (n + 1) / 2 ) )
|
|
ALLOCATE( aux( naux ) )
|
|
!
|
|
! ... copy to upper triangular packed matrix
|
|
!
|
|
ij = 0
|
|
DO j = 1, n
|
|
DO i = 1, j
|
|
ij = ij + 1
|
|
hp(ij) = h(i,j)
|
|
END DO
|
|
END DO
|
|
!
|
|
! ... only the first processor diagonalize the matrix
|
|
!
|
|
IF ( me_pool == root_pool ) THEN
|
|
!
|
|
CALL DSPEV( 21, hp, e, v, ldh, n, aux, naux )
|
|
!
|
|
END IF
|
|
!
|
|
CALL mp_bcast( e, root_pool, intra_pool_comm )
|
|
CALL mp_bcast( v, root_pool, intra_pool_comm )
|
|
!
|
|
DEALLOCATE( aux )
|
|
DEALLOCATE( hp )
|
|
!
|
|
RETURN
|
|
!
|
|
END SUBROUTINE rdiagh_aix
|
|
!
|
|
#else
|
|
!
|
|
!-----------------------------------------------------------------------
|
|
SUBROUTINE rdiagh_lapack( )
|
|
!-----------------------------------------------------------------------
|
|
!
|
|
IMPLICIT NONE
|
|
!
|
|
! ... local variables (LAPACK version)
|
|
!
|
|
INTEGER :: lwork, nb, info
|
|
INTEGER, EXTERNAL :: ILAENV
|
|
! ILAENV returns optimal block size "nb"
|
|
REAL (KIND=DP), ALLOCATABLE :: work(:)
|
|
!
|
|
!
|
|
! ... check for the block size
|
|
!
|
|
nb = ILAENV( 1, 'DSYTRD', 'U', n, - 1, - 1, - 1 )
|
|
!
|
|
IF ( nb < 1 ) nb = MAX( 1, n )
|
|
!
|
|
lwork = ( nb + 3 ) * n
|
|
!
|
|
! ... only the first processor diagonalize the matrix
|
|
!
|
|
IF ( me_pool == root_pool ) THEN
|
|
!
|
|
! ... allocate workspace
|
|
!
|
|
v = h
|
|
!
|
|
ALLOCATE( work( lwork ) )
|
|
!
|
|
CALL DSYEV( 'V', 'U', n, v, ldh, e, work, lwork, info )
|
|
!
|
|
CALL errore( 'rdiagh', 'info =/= 0', ABS( info ) )
|
|
!
|
|
! ... deallocate workspace
|
|
!
|
|
DEALLOCATE( work )
|
|
!
|
|
END IF
|
|
!
|
|
CALL mp_bcast( e, root_pool, intra_pool_comm )
|
|
CALL mp_bcast( v, root_pool, intra_pool_comm )
|
|
!
|
|
RETURN
|
|
!
|
|
END SUBROUTINE rdiagh_lapack
|
|
!
|
|
#endif
|
|
!
|
|
END SUBROUTINE rdiagh
|