test_scalapack_template.F90 3.6 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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
! (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.


  subroutine solve_p&
      &BLAS_CHAR_AND_SY_OR_HE&
      &evd(na, a, sc_desc, ev, z)
    implicit none
#include "../../src/general/precision_kinds.F90"
    integer(kind=ik), intent(in)     :: na
    MATH_DATATYPE(kind=rck), intent(in)       :: a(:,:)
    MATH_DATATYPE(kind=rck), intent(inout)    :: z(:,:)
    real(kind=rk), intent(inout)    :: ev(:)
    integer(kind=ik), intent(in)     :: sc_desc(:)
    integer(kind=ik)                 :: info, lwork, liwork, lrwork
    MATH_DATATYPE(kind=rck), allocatable      :: work(:)
    real(kind=rk), allocatable       :: rwork(:)
    integer, allocatable             :: iwork(:)

    allocate(work(1), iwork(1), rwork(1))

    ! query for required workspace
#ifdef REALCASE
    call p&
         &BLAS_CHAR&
         &syevd('V', 'U', na, a, 1, 1, sc_desc, ev, z, 1, 1, sc_desc, work, -1, iwork, -1, info)
#endif
#ifdef COMPLEXCASE
    call p&
         &BLAS_CHAR&
         &heevd('V', 'U', na, a, 1, 1, sc_desc, ev, z, 1, 1, sc_desc, work, -1, rwork, -1, iwork, -1, info)
#endif
    !  write(*,*) "computed sizes", lwork, liwork, "required sizes ", work(1), iwork(1)
    lwork = work(1)
    liwork = iwork(1)
    deallocate(work, iwork)
    allocate(work(lwork), stat = info)
    allocate(iwork(liwork), stat = info)
#ifdef COMPLEXCASE
    lrwork = rwork(1)
    deallocate(rwork)
    allocate(rwork(lrwork), stat = info)
#endif
    ! the actuall call to the method
#ifdef REALCASE
    call p&
         &BLAS_CHAR&
         &syevd('V', 'U', na, a, 1, 1, sc_desc, ev, z, 1, 1, sc_desc, work, lwork, iwork, liwork, info)
#endif
#ifdef COMPLEXCASE
    call p&
         &BLAS_CHAR&
         &heevd('V', 'U', na, a, 1, 1, sc_desc, ev, z, 1, 1, sc_desc, work, lwork, rwork, lrwork, iwork, liwork, info)
#endif

    deallocate(iwork, work, rwork)
  end subroutine