read_input_parameters.F90 13.8 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
45
!
!    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

46
47
48
  use output_types

  implicit none
49

50
51
52
  type input_options_t
    integer        :: datatype
    type(output_t) :: write_to_file
Andreas Marek's avatar
Andreas Marek committed
53
54
    logical        :: doSolveTridi, do1stage, do2stage, justHelpMessage, &
                      doCholesky, doInvertTrm, doTransposeMultiply
55
56
57
58
59
60
61
  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
62
63
  contains

64
65
    subroutine parse_arguments(obj, command_line_argument, input_options)
      use elpa
66
67
68
69
70
71
72
      use precision
      use output_types

      implicit none

      type(input_options_t) :: input_options
      character(len=128)    :: command_line_argument
73
74
      class(elpa_t), intent(inout) :: obj
      integer :: value, success
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
103
        read(command_line_argument(4:), *) value
        call obj%set("na", value)
104
105
      endif
      if (command_line_argument(1:4) == "nev=") then
106
107
        read(command_line_argument(5:), *) value
        call obj%set("nev", value)
108
109
      endif
      if (command_line_argument(1:5) == "nblk=") then
110
111
        read(command_line_argument(6:), *) value
        call obj%set("nblk", value)
112
113
114
115
116
117
118
119
120
121
122
      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
123
124
125
126
127
128
        value = elpa_int_string_to_value("real_kernel", trim(command_line_argument(15:)), success)
        if (success /= ELPA_OK) then
          print *, "ERROR: Could not parse value for option --real-kernel"
          stop 1
        endif
        call obj%set("real_kernel", value)
129
130
131
      endif

      if (command_line_argument(1:17) == "--complex-kernel=") then
132
133
134
135
136
137
        value = elpa_int_string_to_value("complex_kernel", trim(command_line_argument(18:)), success)
        if (success /= ELPA_OK) then
          print *, "ERROR: Could not parse value for option --complex-kernel"
          stop 1
        endif
        call obj%set("complex_kernel", value)
138
139
140
      endif

      if (command_line_argument(1:9) == "--use-qr=") then
141
142
        read(command_line_argument(10:), *) value
        call obj%set("qr", value)
143
144
145
      endif

      if (command_line_argument(1:10) == "--use-gpu=") then
146
147
        read(command_line_argument(11:), *) value
        call obj%set("gpu", value)
148
149
150
151
152
153
154
      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
155
156
157
          input_options%doCholesky=.true.
          input_options%doInvertTrm=.true.
          input_options%doTransposeMultiply=.true.
158
159
160
161
        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
162
163
164
          input_options%doCholesky=.false.
          input_options%doInvertTrm=.false.
          input_options%doTransposeMultiply=.false.
165
166
167
168
        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
169
170
171
          input_options%doCholesky=.false.
          input_options%doInvertTrm=.false.
          input_options%doTransposeMultiply=.false.
172
173
174
175
        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
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
          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.
200
201
        else
           print *,"unknown test specified"
Pavel Kus's avatar
Pavel Kus committed
202
           stop 1
203
204
205
206
207
        endif
      endif

    end subroutine

208
209
    subroutine read_input_parameters_general(obj, input_options)
      use elpa
210
211
212
213
214
215
      use ELPA_utilities, only : error_unit
      use precision
      use elpa_mpi
      use output_types
      implicit none

216
      class(elpa_t), intent(inout) :: obj
217
218
219
220
221
222
223
224
      type(input_options_t)         :: input_options

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

      ! default parameters
      input_options%datatype = 1
225
226
227
      call obj%set("na", 4000)
      call obj%set("nev", 1500)
      call obj%set("nblk", 16)
228
229
230
231
232
233
234

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

      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(obj, 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(obj, 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(obj, 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(obj, 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(obj, 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(obj, 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(obj, 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(obj, 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(obj, 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(obj, arg8, input_options)
327
328
329
330
331

        endif

      endif

332
      if (obj%is_set("qr") == 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)
340
      use ELPA_utilities, only : error_unit
341
      use precision
342
      use elpa_mpi
343
      use output_types
Andreas Marek's avatar
Andreas Marek committed
344
      implicit none
345

346
      integer(kind=ik), intent(out) :: na, nev, nblk
347
348

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

350
      ! Command line arguments
351
      character(len=128)            :: arg1, arg2, arg3, arg4, arg5
352
      integer(kind=ik)              :: mpierr
Andreas Marek's avatar
Andreas Marek committed
353

354
355
356
357
      ! default parameters
      na = 4000
      nev = 1500
      nblk = 16
358
359
      write_to_file%eigenvectors = .false.
      write_to_file%eigenvalues  = .false.
Andreas Marek's avatar
Andreas Marek committed
360

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

      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

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

Andreas Marek's avatar
Andreas Marek committed
394
395
      endif

396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
      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
421
422
423
    end subroutine

end module