read_input_parameters.F90 14.1 KB
Newer Older
Andreas Marek's avatar
Andreas Marek committed
1
2
3
4
5
!    This file is part of ELPA.
!
!    The ELPA library was originally created by the ELPA consortium,
!    consisting of the following organizations:
!
6
7
!    - Max Planck Computing and Data Facility (MPCDF), formerly known as
!      Rechenzentrum Garching der Max-Planck-Gesellschaft (RZG),
Andreas Marek's avatar
Andreas Marek committed
8
9
10
11
12
13
14
15
16
17
18
19
!    - 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:
20
!    http://elpa.mpcdf.mpg.de/
Andreas Marek's avatar
Andreas Marek committed
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
!
!    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 mod_read_input_parameters
45
46
  use elpa
  use elpa_utilities, only : error_unit
47
  use output_types
48
  use precision
49
50
51
52

  implicit none
  type input_options_t
    integer        :: datatype
53
    integer        :: na, nev, nblk
54
    type(output_t) :: write_to_file
55
56
57
    integer        :: this_real_kernel, this_complex_kernel
    logical        :: realKernelIsSet, complexKernelIsSet
    integer        :: useQrIsSet, useGPUIsSet
Andreas Marek's avatar
Andreas Marek committed
58
59
    logical        :: doSolveTridi, do1stage, do2stage, justHelpMessage, &
                      doCholesky, doInvertTrm, doTransposeMultiply
60
61
62
63
64
65
66
  end type

  interface read_input_parameters
    module procedure read_input_parameters_general
    module procedure read_input_parameters_traditional
  end interface

Andreas Marek's avatar
Andreas Marek committed
67
68
  contains

69
    subroutine parse_arguments(command_line_argument, input_options)
70
71
72
73
      implicit none

      type(input_options_t) :: input_options
      character(len=128)    :: command_line_argument
74
      integer               :: error
75
76

      if (command_line_argument == "--help") then
77
78
79
80
        print *,"usage: elpa_tests [--help] [datatype={real|complex}] [na=number] [nev=number] "
        print *,"                  [nblk=size of block cyclic distribution] [--output_eigenvalues]"
        print *,"                  [--output_eigenvectors] [--real-kernel=name_of_kernel]"
        print *,"                  [--complex-kernel=name_of_kernel] [--use-gpu={0|1}]"
Andreas Marek's avatar
Andreas Marek committed
81
82
        print *,"                  [--use-qr={0,1}] [--tests={all|solve-tridi|1stage|2stage|cholesky&
            &|invert-triangular|transpose-mulitply}]"
Andreas Marek's avatar
Andreas Marek committed
83
84
        input_options%justHelpMessage=.true.
        return
85
86
87
88
89
90
91
92
93
94
95
      endif


      if (command_line_argument(1:11) == "--datatype=") then
        if (command_line_argument(12:15) == "real") then
          input_options%datatype=1
        else
          if (command_line_argument(12:18) == "complex") then
            input_options%datatype=2
          else
            print *,"datatype unknown! use either --datatype=real or --datatpye=complex"
Pavel Kus's avatar
Pavel Kus committed
96
            stop 1
97
98
99
100
101
          endif
        endif
      endif

      if (command_line_argument(1:3) == "na=") then
102
        read(command_line_argument(4:), *) input_options%na
103
104
      endif
      if (command_line_argument(1:4) == "nev=") then
105
        read(command_line_argument(5:), *) input_options%nev
106
107
      endif
      if (command_line_argument(1:5) == "nblk=") then
108
        read(command_line_argument(6:), *) input_options%nblk
109
110
111
112
113
114
115
116
117
118
119
      endif

      if (command_line_argument(1:21)   == "--output_eigenvectors") then
        input_options%write_to_file%eigenvectors = .true.
      endif

      if (command_line_argument(1:20)   == "--output_eigenvalues") then
        input_options%write_to_file%eigenvalues = .true.
      endif

      if (command_line_argument(1:14) == "--real-kernel=") then
120
121
122
        input_options%this_real_kernel = elpa_int_string_to_value("real_kernel", command_line_argument(15:), error)
        if (error /= ELPA_OK) then
          print *, "Invalid argument for --real-kernel"
123
124
          stop 1
        endif
125
126
        print *,"Setting ELPA2 real kernel to ", elpa_int_value_to_string("real_kernel", input_options%this_real_kernel)
        input_options%realKernelIsSet = .true.
127
128
129
      endif

      if (command_line_argument(1:17) == "--complex-kernel=") then
130
131
132
        input_options%this_complex_kernel = elpa_int_string_to_value("complex_kernel", command_line_argument(18:), error)
        if (error /= ELPA_OK) then
          print *, "Invalid argument for --complex-kernel"
133
134
          stop 1
        endif
135
136
        print *,"Setting ELPA2 complex kernel to ", elpa_int_value_to_string("complex_kernel", input_options%this_complex_kernel)
        input_options%complexKernelIsSet = .true.
137
138
139
      endif

      if (command_line_argument(1:9) == "--use-qr=") then
140
        read(command_line_argument(10:), *) input_options%useQrIsSet
141
142
143
      endif

      if (command_line_argument(1:10) == "--use-gpu=") then
144
        read(command_line_argument(11:), *) input_options%useGPUIsSet
145
146
147
148
149
150
151
      endif

      if (command_line_argument(1:8) == "--tests=") then
        if (command_line_argument(9:11) == "all") then
          input_options%doSolveTridi=.true.
          input_options%do1stage=.true.
          input_options%do2stage=.true.
Andreas Marek's avatar
Andreas Marek committed
152
153
154
          input_options%doCholesky=.true.
          input_options%doInvertTrm=.true.
          input_options%doTransposeMultiply=.true.
155
156
157
158
        else if (command_line_argument(9:19) == "solve-tride") then
          input_options%doSolveTridi=.true.
          input_options%do1stage=.false.
          input_options%do2stage=.false.
Andreas Marek's avatar
Andreas Marek committed
159
160
161
          input_options%doCholesky=.false.
          input_options%doInvertTrm=.false.
          input_options%doTransposeMultiply=.false.
162
163
164
165
        else if (command_line_argument(9:14) == "1stage") then
          input_options%doSolveTridi=.false.
          input_options%do1stage=.true.
          input_options%do2stage=.false.
Andreas Marek's avatar
Andreas Marek committed
166
167
168
          input_options%doCholesky=.false.
          input_options%doInvertTrm=.false.
          input_options%doTransposeMultiply=.false.
169
170
171
172
        else if (command_line_argument(9:14) == "2stage") then
          input_options%doSolveTridi=.false.
          input_options%do1stage=.false.
          input_options%do2stage=.true.
Andreas Marek's avatar
Andreas Marek committed
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
          input_options%doCholesky=.false.
          input_options%doInvertTrm=.false.
          input_options%doTransposeMultiply=.false.
        else if (command_line_argument(9:16) == "cholesky") then
          input_options%doSolveTridi=.false.
          input_options%do1stage=.false.
          input_options%do2stage=.false.
          input_options%doCholesky=.true.
          input_options%doInvertTrm=.false.
          input_options%doTransposeMultiply=.false.
        else if (command_line_argument(9:25) == "invert-triangular") then
          input_options%doSolveTridi=.false.
          input_options%do1stage=.false.
          input_options%do2stage=.false.
          input_options%doCholesky=.false.
          input_options%doInvertTrm=.true.
          input_options%doTransposeMultiply=.false.
        else if (command_line_argument(9:26) == "transpose-multiply") then
          input_options%doSolveTridi=.false.
          input_options%do1stage=.false.
          input_options%do2stage=.false.
          input_options%doCholesky=.false.
          input_options%doInvertTrm=.false.
          input_options%doTransposeMultiply=.true.
197
198
        else
           print *,"unknown test specified"
Pavel Kus's avatar
Pavel Kus committed
199
           stop 1
200
201
202
203
204
        endif
      endif

    end subroutine

205
    subroutine read_input_parameters_general(input_options)
206
207
208
209
210
211
      implicit none

      type(input_options_t)         :: input_options

      ! Command line arguments
      character(len=128)            :: arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9, arg10
212
      integer(kind=ik)              :: mpierr
213
214
215

      ! default parameters
      input_options%datatype = 1
216
217
218
      input_options%na = 4000
      input_options%nev = 1500
      input_options%nblk = 16
219
220
221
222

      input_options%write_to_file%eigenvectors = .false.
      input_options%write_to_file%eigenvalues  = .false.

223
224
      input_options%this_real_kernel = ELPA_2STAGE_REAL_DEFAULT
      input_options%this_complex_kernel = ELPA_2STAGE_COMPLEX_DEFAULT
225
226
227
228
229
230
231
      input_options%realKernelIsSet = .false.
      input_options%complexKernelIsSet = .false.

      input_options%useQrIsSet = 0

      input_options%useGPUIsSet = 0

232
233
234
      input_options%do1Stage = .true.
      input_options%do2Stage = .true.
      input_options%doSolveTridi = .true.
Andreas Marek's avatar
Andreas Marek committed
235
236
237
      input_options%doCholesky=.true.
      input_options%doInvertTrm=.true.
      input_options%doTransposeMultiply=.true.
Andreas Marek's avatar
Andreas Marek committed
238
239
      input_options%justHelpMessage=.false.

240
241
242
243
244
245
246
247
248
249
      ! test na=1500 nev=50 nblk=16 --help --kernel --output_eigenvectors --output_eigenvalues
      if (COMMAND_ARGUMENT_COUNT() .gt. 8) then
        write(error_unit, '(a,i0,a)') "Invalid number (", COMMAND_ARGUMENT_COUNT(), ") of command line arguments!"
        stop 1
      endif

      if (COMMAND_ARGUMENT_COUNT() .gt. 0) then

        call get_COMMAND_ARGUMENT(1, arg1)

250
        call parse_arguments(arg1, input_options)
251
252
253
254
255
256
257



        if (COMMAND_ARGUMENT_COUNT() .ge. 2) then
          ! argument 2
          call get_COMMAND_ARGUMENT(2, arg2)

258
          call parse_arguments(arg2, input_options)
259
260
261
262
263
264
265
        endif

        ! argument 3
        if (COMMAND_ARGUMENT_COUNT() .ge. 3) then

          call get_COMMAND_ARGUMENT(3, arg3)

266
          call parse_arguments(arg3, input_options)
267
268
269
270
271
272
273
        endif

        ! argument 4
        if (COMMAND_ARGUMENT_COUNT() .ge. 4) then

          call get_COMMAND_ARGUMENT(4, arg4)

274
          call parse_arguments(arg4, input_options)
275
276
277
278
279
280
281
282

        endif

        ! argument 5
        if (COMMAND_ARGUMENT_COUNT() .ge. 5) then

          call get_COMMAND_ARGUMENT(5, arg5)

283
          call parse_arguments(arg5, input_options)
284
285
286
287
288
289
290
        endif

        ! argument 6
        if (COMMAND_ARGUMENT_COUNT() .ge. 6) then

          call get_COMMAND_ARGUMENT(6, arg6)

291
          call parse_arguments(arg6, input_options)
292
293
294
295
296
297
298
        endif

        ! argument 7
        if (COMMAND_ARGUMENT_COUNT() .ge. 7) then

          call get_COMMAND_ARGUMENT(7, arg7)

299
          call parse_arguments(arg7, input_options)
300
301
302
303
304
305
306
307

        endif

        ! argument 8
        if (COMMAND_ARGUMENT_COUNT() .ge. 8) then

          call get_COMMAND_ARGUMENT(8, arg8)

308
          call parse_arguments(arg8, input_options)
309
310
311
312
313
314
315
316

        endif

        ! argument 9
        if (COMMAND_ARGUMENT_COUNT() .ge. 9) then

          call get_COMMAND_ARGUMENT(9, arg9)

317
          call parse_arguments(arg8, input_options)
318
319
320
321
322
323
324
325

        endif

        ! argument 10
        if (COMMAND_ARGUMENT_COUNT() .ge. 10) then

          call get_COMMAND_ARGUMENT(10, arg10)

326
          call parse_arguments(arg8, input_options)
327
328
329
330
331

        endif

      endif

332
      if (input_options%useQrIsSet .eq. 1 .and. input_options%datatype .eq. 2) then
333
334
335
336
337
338
339
        print *,"You cannot use QR-decomposition in complex case"
        stop 1
      endif

    end subroutine

    subroutine read_input_parameters_traditional(na, nev, nblk, write_to_file)
Andreas Marek's avatar
Andreas Marek committed
340
      implicit none
341

342
      integer(kind=ik), intent(out) :: na, nev, nblk
343
344

      type(output_t), intent(out)   :: write_to_file
Andreas Marek's avatar
Andreas Marek committed
345

346
      ! Command line arguments
347
      character(len=128)            :: arg1, arg2, arg3, arg4, arg5
348
      integer(kind=ik)              :: mpierr
Andreas Marek's avatar
Andreas Marek committed
349

350
351
352
353
      ! default parameters
      na = 4000
      nev = 1500
      nblk = 16
354
355
      write_to_file%eigenvectors = .false.
      write_to_file%eigenvalues  = .false.
Andreas Marek's avatar
Andreas Marek committed
356

357
      if (.not. any(COMMAND_ARGUMENT_COUNT() == [0, 3, 4, 5])) then
358
        write(error_unit, '(a,i0,a)') "Invalid number (", COMMAND_ARGUMENT_COUNT(), ") of command line arguments!"
359
360
        write(error_unit, *) "Expected: program [ [matrix_size num_eigenvalues block_size] &
            ""output_eigenvalues"" ""output_eigenvectors""]"
361
362
        stop 1
      endif
Andreas Marek's avatar
Andreas Marek committed
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382

      if (COMMAND_ARGUMENT_COUNT() == 3) then
        call GET_COMMAND_ARGUMENT(1, arg1)
        call GET_COMMAND_ARGUMENT(2, arg2)
        call GET_COMMAND_ARGUMENT(3, arg3)

        read(arg1, *) na
        read(arg2, *) nev
        read(arg3, *) nblk
      endif

      if (COMMAND_ARGUMENT_COUNT() == 4) then
        call GET_COMMAND_ARGUMENT(1, arg1)
        call GET_COMMAND_ARGUMENT(2, arg2)
        call GET_COMMAND_ARGUMENT(3, arg3)
        call GET_COMMAND_ARGUMENT(4, arg4)
        read(arg1, *) na
        read(arg2, *) nev
        read(arg3, *) nblk

383
        if (arg4 .eq. "output_eigenvalues") then
384
          write_to_file%eigenvalues = .true.
385
        else
386
          write(error_unit, *) "Invalid value for output flag! Must be ""output_eigenvalues"" or omitted"
387
388
389
          stop 1
        endif

Andreas Marek's avatar
Andreas Marek committed
390
391
      endif

392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
      if (COMMAND_ARGUMENT_COUNT() == 5) then
        call GET_COMMAND_ARGUMENT(1, arg1)
        call GET_COMMAND_ARGUMENT(2, arg2)
        call GET_COMMAND_ARGUMENT(3, arg3)
        call GET_COMMAND_ARGUMENT(4, arg4)
        call GET_COMMAND_ARGUMENT(5, arg5)
        read(arg1, *) na
        read(arg2, *) nev
        read(arg3, *) nblk

        if (arg4 .eq. "output_eigenvalues") then
          write_to_file%eigenvalues = .true.
        else
          write(error_unit, *) "Invalid value for output flag! Must be ""output_eigenvalues"" or omitted"
          stop 1
        endif

        if (arg5 .eq. "output_eigenvectors") then
          write_to_file%eigenvectors = .true.
        else
          write(error_unit, *) "Invalid value for output flag! Must be ""output_eigenvectors"" or omitted"
          stop 1
        endif

      endif
Andreas Marek's avatar
Andreas Marek committed
417
418
419
    end subroutine

end module