test_scalapack.F90 3.01 KB
Newer Older
Pavel Kus's avatar
Pavel Kus committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
! (c) Copyright Pavel Kus, 2017, MPCDF
!
!    This file is part of ELPA.
!
!    The ELPA library was originally created by the ELPA consortium,
!    consisting of the following organizations:
!
!    - Max Planck Computing and Data Facility (MPCDF), formerly known as
!      Rechenzentrum Garching der Max-Planck-Gesellschaft (RZG),
!    - Bergische Universität Wuppertal, Lehrstuhl für angewandte
!      Informatik,
!    - Technische Universität München, Lehrstuhl für Informatik mit
!      Schwerpunkt Wissenschaftliches Rechnen ,
!    - Fritz-Haber-Institut, Berlin, Abt. Theorie,
!    - Max-Plack-Institut für Mathematik in den Naturwissenschaften,
!      Leipzig, Abt. Komplexe Strukutren in Biologie und Kognition,
!      and
!    - IBM Deutschland GmbH
!
!
!    More information can be found here:
!    http://elpa.mpcdf.mpg.de/
!
!    ELPA is free software: you can redistribute it and/or modify
!    it under the terms of the version 3 of the license of the
!    GNU Lesser General Public License as published by the Free
!    Software Foundation.
!
!    ELPA is distributed in the hope that it will be useful,
!    but WITHOUT ANY WARRANTY; without even the implied warranty of
!    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
!    GNU Lesser General Public License for more details.
!
!    You should have received a copy of the GNU Lesser General Public License
!    along with ELPA.  If not, see <http://www.gnu.org/licenses/>
!
!    ELPA reflects a substantial effort on the part of the original
!    ELPA consortium, and we ask you to respect the spirit of the
!    license that we chose: i.e., please contribute any changes you
!    may have back to the original ELPA library distribution, and keep
!    any derivatives of ELPA under the same license that we chose for
!    the original distribution, the GNU Lesser General Public License.

#include "config-f90.h"

module test_scalapack
  use test_util

  interface solve_scalapack_all
    module procedure solve_pdsyevd
  end interface  

contains

  subroutine solve_pdsyevd(na, a, sc_desc, ev, z)
    implicit none 
    integer(kind=ik), intent(in)     :: na
    real(kind=rk8), intent(in)       :: a(:,:)
    real(kind=rk8), intent(inout)    :: z(:,:), ev(:)
    integer(kind=ik), intent(in)     :: sc_desc(:)
    integer(kind=ik)                 :: info, lwork, liwork
    real(kind=rk8), allocatable      :: work(:)
    integer, allocatable             :: iwork(:) 
64

Pavel Kus's avatar
Pavel Kus committed
65
66
67
68
69
70
71
    allocate(work(1), iwork(1))

    ! query for required workspace
    call pdsyevd('V', 'U', na, a, 1, 1, sc_desc, ev, z, 1, 1, sc_desc, work, -1, iwork, -1, info)
    !  write(*,*) "computed sizes", lwork, liwork, "required sizes ", work(1), iwork(1)
    lwork = work(1)
    liwork = iwork(1)
72

Pavel Kus's avatar
Pavel Kus committed
73
74
75
    deallocate(work, iwork)
    allocate(work(lwork), stat = info)
    allocate(iwork(liwork), stat = info)
76

Pavel Kus's avatar
Pavel Kus committed
77
78
79
80
81
82
83
84
    ! the actuall call to the method
    call pdsyevd('V', 'U', na, a, 1, 1, sc_desc, ev, z, 1, 1, sc_desc, work, lwork, iwork, liwork, info)

    deallocate(iwork)
    deallocate(work)
  end subroutine

end module