source: palm/trunk/SOURCE/shared_memory_io_mod.f90 @ 4753

Last change on this file since 4753 was 4629, checked in by raasch, 4 years ago

support for MPI Fortran77 interface (mpif.h) removed

  • Property svn:keywords set to Id
File size: 46.4 KB
RevLine 
[4534]1!> @file shared_memory_io_mod.f90
[4591]2!--------------------------------------------------------------------------------------------------!
3! This file is part of the PALM model system.
[4534]4!
[4591]5! PALM is free software: you can redistribute it and/or modify it under the terms of the GNU General
6! Public License as published by the Free Software Foundation, either version 3 of the License, or
7! (at your option) any later version.
[4534]8!
[4591]9! PALM is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the
10! implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
11! Public License for more details.
[4534]12!
[4591]13! You should have received a copy of the GNU General Public License along with PALM. If not, see
14! <http://www.gnu.org/licenses/>.
[4534]15!
16! Copyright 1997-2020 Leibniz Universitaet Hannover
[4591]17!--------------------------------------------------------------------------------------------------!
[4534]18!
[4591]19!
[4534]20! Current revisions:
21! -----------------
[4591]22!
23!
[4534]24! Former revisions:
[4620]25! -----------------
[4534]26! $Id: shared_memory_io_mod.f90 4629 2020-07-29 09:37:56Z raasch $
[4629]27! support for MPI Fortran77 interface (mpif.h) removed
28!
29! 4628 2020-07-29 07:23:03Z raasch
[4628]30! extensions required for MPI-I/O of particle data to restart files
31!
32! 4620 2020-07-22 14:11:16Z raasch
[4620]33! bugfix: variable definition changed
[4534]34!
[4620]35! 4618 2020-07-22 11:21:08Z raasch
[4618]36! unused variable removed
37!
38!
[4617]39! Additions for cyclic fill mode
40!
41!
[4591]42! File re-formatted to follow the PALM coding standard
[4534]43!
[4591]44!
45! Initial version (Klaus Ketelsen)
[4534]46!
[4591]47!
48!
[4534]49! Description:
50! ------------
[4591]51!> Handle MPI-IO or NetCDF-IO shared memory arrays.
52!> This module performs the organization of new communicators, adapted PE-grids and allocation of
53!> shared memory arrays. The IO itself is not done here.
54!--------------------------------------------------------------------------------------------------!
[4534]55 MODULE shared_memory_io_mod
56
57#if defined( __parallel )
58    USE MPI
59#endif
60
61    USE, INTRINSIC ::  ISO_C_BINDING
62
63    USE control_parameters,                                                                        &
[4591]64        ONLY: maximum_grid_level,                                                                  &
65              message_string,                                                                      &
66              mg_switch_to_pe0_level
[4534]67
[4591]68
[4534]69    USE indices,                                                                                   &
[4591]70        ONLY: nbgp,                                                                                &
71              nnx,                                                                                 &
72              nny,                                                                                 &
73              nnz,                                                                                 &
74              nx,                                                                                  &
75              nxl,                                                                                 &
76              nxlg,                                                                                &
77              nxr,                                                                                 &
78              nxrg,                                                                                &
79              ny,                                                                                  &
80              nyn,                                                                                 &
81              nyng,                                                                                &
82              nys,                                                                                 &
83              nysg,                                                                                &
84              nzb,                                                                                 &
85              nzt
[4534]86
87    USE kinds,                                                                                     &
[4617]88        ONLY: dp,                                                                                  &
[4628]89              idp,                                                                                 &
90              isp,                                                                                 &
[4617]91              iwp,                                                                                 &
92              sp,                                                                                  &
[4591]93              wp
[4534]94
95    USE pegrid,                                                                                    &
[4591]96        ONLY: comm1dx,                                                                             &
97              comm1dy,                                                                             &
98              comm2d,                                                                              &
99              ierr,                                                                                &
100              myid,                                                                                &
101              myidx,                                                                               &
102              myidy,                                                                               &
103              npex,                                                                                &
104              npey,                                                                                &
105              numprocs,                                                                            &
106              pdims,                                                                               &
107              pleft,                                                                               &
108              pnorth,                                                                              &
109              pright,                                                                              &
110              psouth,                                                                              &
111              sendrecvcount_xy
112
[4534]113#if defined( __parallel )
114    USE pegrid,                                                                                    &
[4591]115        ONLY: pcoord,                                                                              &
116              reorder
[4534]117#endif
118
119    IMPLICIT NONE
120
121    PRIVATE
122
123    SAVE
124
125!
[4617]126!-- Type to store information about the domain decomposition grid
127    TYPE, PUBLIC ::  domain_decomposition_grid_features  !<
[4534]128
[4617]129       INTEGER(iwp) ::  comm2d    !<
130       INTEGER(iwp) ::  myid      !<
131       INTEGER(iwp) ::  nnx       !<
132       INTEGER(iwp) ::  nny       !<
133       INTEGER(iwp) ::  nx        !<
134       INTEGER(iwp) ::  nxl       !<
135       INTEGER(iwp) ::  nxr       !<
136       INTEGER(iwp) ::  ny        !<
137       INTEGER(iwp) ::  nyn       !<
138       INTEGER(iwp) ::  nys       !<
139       INTEGER(iwp) ::  numprocs  !<
[4534]140
[4617]141       CONTAINS
[4591]142
[4617]143          PROCEDURE, PASS(this), PUBLIC :: activate_grid_from_this_class
144          PROCEDURE, PASS(this), PUBLIC :: save_grid_into_this_class
[4591]145
[4617]146    END TYPE domain_decomposition_grid_features
[4591]147
[4534]148!
149!-- Class definition for shared memory instances.
150!-- For every use of shared memory IO, one instance of this class is created.
[4591]151    TYPE, PUBLIC ::  sm_class  !<
[4534]152
[4591]153       INTEGER(iwp) ::  nr_io_pe_per_node = 2         !< typical configuration, 2 sockets per node
154       LOGICAL      ::  no_shared_Memory_in_this_run  !<
[4617]155
156       INTEGER(iwp) ::  comm_model            !< communicator of this model run
[4534]157!
158!--    Variables for the shared memory communicator
[4617]159       INTEGER(iwp), PUBLIC ::  comm_shared   !< communicator for processes with shared array
[4591]160       INTEGER(iwp), PUBLIC ::  sh_npes       !<
161       INTEGER(iwp), PUBLIC ::  sh_rank       !<
[4534]162
163       LOGICAL, PUBLIC ::  iam_io_pe = .TRUE.  !< This PE is an IO-PE
164!
165!--    Variables for the I/O virtual grid
[4591]166       INTEGER(iwp), PUBLIC ::  comm_io  !< Communicator for all IO processes
167       INTEGER(iwp), PUBLIC ::  io_npes  !<
168       INTEGER(iwp), PUBLIC ::  io_rank  !<
[4534]169!
170!--    Variables for the node local communicator
[4591]171       INTEGER(iwp) ::  comm_node          !< Communicator for all processes of current node
172       INTEGER(iwp) ::  io_pe_global_rank  !<
173       INTEGER(iwp) ::  n_npes             !<
174       INTEGER(iwp) ::  n_rank             !<
[4534]175
[4617]176       TYPE(domain_decomposition_grid_features), PUBLIC ::  io_grid  !< io grid features, depending on reading from prerun or restart run
[4534]177
178
[4617]179       CONTAINS
180
181          PRIVATE
182
183          PROCEDURE, PASS(this), PUBLIC ::  is_sm_active
184          PROCEDURE, PASS(this), PUBLIC ::  sm_adjust_outer_boundary
185          PROCEDURE, PASS(this), PUBLIC ::  sm_free_shared
186          PROCEDURE, PASS(this), PUBLIC ::  sm_init_comm
187          PROCEDURE, PASS(this), PUBLIC ::  sm_init_part
188          PROCEDURE, PASS(this), PUBLIC ::  sm_node_barrier
[4534]189#if defined( __parallel )
[4617]190          PROCEDURE, PASS(this), PUBLIC ::  sm_allocate_shared_1d_64
191          PROCEDURE, PASS(this), PUBLIC ::  sm_allocate_shared_1d_32
192          PROCEDURE, PASS(this), PUBLIC ::  sm_allocate_shared_1di
193          PROCEDURE, PASS(this), PUBLIC ::  sm_allocate_shared_2d_64
194          PROCEDURE, PASS(this), PUBLIC ::  sm_allocate_shared_2d_32
195          PROCEDURE, PASS(this), PUBLIC ::  sm_allocate_shared_2di
196          PROCEDURE, PASS(this), PUBLIC ::  sm_allocate_shared_3d_64
197          PROCEDURE, PASS(this), PUBLIC ::  sm_allocate_shared_3d_32
[4628]198          PROCEDURE, PASS(this), PUBLIC ::  sm_allocate_shared_3di_32
199          PROCEDURE, PASS(this), PUBLIC ::  sm_allocate_shared_3di_64
[4534]200
[4617]201          GENERIC, PUBLIC ::  sm_allocate_shared =>                                                &
[4628]202                                              sm_allocate_shared_1d_64,  sm_allocate_shared_1d_32, &
203                                              sm_allocate_shared_2d_64,  sm_allocate_shared_2d_32, &
204                                              sm_allocate_shared_2di,    sm_allocate_shared_3d_64, &
205                                              sm_allocate_shared_3d_32,  sm_allocate_shared_1di,   &
206                                              sm_allocate_shared_3di_32, sm_allocate_shared_3di_64
[4534]207#endif
208    END TYPE sm_class
209
210
[4591]211 CONTAINS
[4534]212
213
214!--------------------------------------------------------------------------------------------------!
215! Description:
216! ------------
217!> Create the communicator for shared memory groups and IO-PEs.
218!> Setup the grid for shared memory IO.
219!--------------------------------------------------------------------------------------------------!
[4617]220 SUBROUTINE sm_init_comm( this, sm_active, comm_input )
[4534]221
222    IMPLICIT NONE
223
[4617]224    CLASS(sm_class), INTENT(INOUT) ::  this        !< pointer to access internal variables of this call
225    INTEGER, INTENT(IN), OPTIONAL  ::  comm_input  !< main model communicator (comm2d) can optional be set
[4534]226
227#if defined( __parallel )
[4617]228    INTEGER ::  color
229    INTEGER ::  max_n_npes  !< maximum number of PEs/node
[4534]230#endif
231
[4617]232    LOGICAL, INTENT(IN) ::  sm_active  !< flag to activate shared-memory IO
[4534]233
[4617]234    IF ( PRESENT( comm_input ) )  THEN
235       this%comm_model = comm_input
236    ELSE
237       this%comm_model = comm2d
238    ENDIF
[4534]239
240    this%no_shared_memory_in_this_run = .NOT. sm_active
[4617]241    this%comm_io = this%comm_model      ! preset in case of non shared-memory-IO
[4534]242
243    IF ( this%no_shared_memory_in_this_run )  THEN
244       this%iam_io_pe = .TRUE.
245       RETURN
246    ENDIF
247
248#if defined( __parallel )
249!
250!-- Determine, how many MPI threads are running on a node
251    this%iam_io_pe = .FALSE.
[4617]252    CALL MPI_COMM_SPLIT_TYPE( this%comm_model, MPI_COMM_TYPE_SHARED, 0, MPI_INFO_NULL,             &
253                              this%comm_node, ierr )
[4534]254    CALL MPI_COMM_SIZE( this%comm_node, this%n_npes, ierr )
255    CALL MPI_COMM_RANK( this%comm_node, this%n_rank, ierr )
256
[4617]257    CALL MPI_ALLREDUCE( this%n_npes, max_n_npes, 1, MPI_INTEGER, MPI_MAX, this%comm_model, ierr )
[4534]258!
259!-- Decide, if the configuration can run with shared-memory IO
260    IF ( max_n_npes > 64 )  THEN
261!
262!--    Special configuration on the HLRN-IV system with 4 shared memory blocks/node
263       this%nr_io_pe_per_node = 4
264
265    ELSEIF ( max_n_npes <= 32 )  THEN
266!
267!--    No shared memory IO with less than 32 threads/node
268       this%no_shared_memory_in_this_run = .TRUE.
269       this%iam_io_pe = .TRUE.
270       RETURN
271    ENDIF
272
273!
274!-- No shared memory IO with small setups
275    IF ( nx < 24  .OR.  ny < 24 )  THEN
276       this%no_shared_memory_in_this_run = .TRUE.
277       this%iam_io_pe = .TRUE.
278       RETURN
279    ENDIF
280
281!
282!-- Divide a node into shared memory groups, depending on the virtual x-y grid
283    CALL compute_color( color )
284!
285!-- If no shared memory IO possible, nothing is left to be done here.
286    IF ( this%no_shared_memory_in_this_run )  RETURN
287
288!
289!-- Setup the shared memory area
290    CALL MPI_COMM_SPLIT( this%comm_node, color, 0, this%comm_shared, ierr )
291    CALL MPI_COMM_SIZE( this%comm_shared, this%sh_npes, ierr )
292    CALL MPI_COMM_RANK( this%comm_shared, this%sh_rank, ierr )
293
294!
295!-- Setup the communicator across the nodes depending on the shared memory rank.
296!-- All threads with shared memory rank 0 will be I/O threads.
297    color = this%sh_rank
[4617]298    CALL MPI_COMM_SPLIT( this%comm_model, color, 0, this%comm_io, ierr )
[4534]299
300    IF ( this%comm_io /= MPI_COMM_NULL )  THEN
301       CALL MPI_COMM_SIZE( this%comm_io, this%io_npes, ierr )
302       CALL MPI_COMM_RANK( this%comm_io, this%io_rank, ierr )
303    ELSE
304       this%io_npes = -1
305       this%io_rank = -1
306    ENDIF
307
308    IF ( this%sh_rank == 0 )  THEN
309       this%iam_io_pe = .TRUE.
310       this%io_pe_global_rank = myid
311    ENDIF
312    CALL MPI_BCAST( this%io_pe_global_rank, 1, MPI_INTEGER, 0, this%comm_shared, ierr )
313
314#else
315    this%iam_io_pe = .TRUE.
316#endif
317
[4617]318!      write(9,'(a,8i7)') ' end of sm_init_comm ',this%sh_rank,this%sh_npes,this%io_rank,this%io_npes,this%io_pe_global_rank
319!      write(9,*) 'This process is IO Process ',this%iam_io_pe
320
[4534]321#if defined( __parallel )
322 CONTAINS
323
324 SUBROUTINE compute_color( color )
325
326    IMPLICIT NONE
327
[4591]328    INTEGER(iwp), INTENT(OUT) ::  color  !<
[4534]329
[4591]330    INTEGER(iwp) ::  group_start    !<
331    INTEGER(iwp) ::  my_color       !<
332    INTEGER(iwp) ::  n              !<
333    INTEGER(iwp) ::  pe             !<
334    INTEGER(iwp) ::  sh_group_size  !<
[4534]335
[4591]336    INTEGER(iwp), DIMENSION(4,0:this%n_npes-1) ::  local_dim_s   !<
337    INTEGER(iwp), DIMENSION(4,0:this%n_npes-1) ::  local_dim_r   !<
[4534]338
[4617]339    TYPE(domain_decomposition_grid_features), DIMENSION(32) ::  node_grid  !<
[4534]340
[4536]341!
[4591]342!-- No shared memory I/O on one node jobs
[4536]343    IF ( numprocs < this%n_npes )  THEN
344       this%no_shared_memory_in_this_run = .TRUE.
[4534]345       RETURN
346    ENDIF
347
348    local_dim_s = 0
349    local_dim_s(1,this%n_rank) = nxl
350    local_dim_s(2,this%n_rank) = nxr
351    local_dim_s(3,this%n_rank) = nys
352    local_dim_s(4,this%n_rank) = nyn
353
354    node_grid%nyn = -1
355!
356!-- Distribute the x-y layout of all cores of a node to all node processes
357    CALL MPI_ALLREDUCE( local_dim_s, local_dim_r, SIZE( local_dim_s ), MPI_INTEGER, MPI_SUM,       &
358                        this%comm_node, ierr )
359    sh_group_size = ( max_n_npes + this%nr_io_pe_per_node - 1 ) / this%nr_io_pe_per_node
360
361    pe       = 0
362    my_color = 1  ! color is used to split the shared memory communicator into a communicator for
363                  ! io groups
364    group_start = pe
365    node_grid(my_color)%nxl = local_dim_r(1,group_start)
366    node_grid(my_color)%nxr = local_dim_r(2,group_start)
367    node_grid(my_color)%nys = local_dim_r(3,group_start)
368
369    DO  n = 1, this%n_npes-1
370
371       pe =  n
372       IF ( n > 0  .AND.  MOD( n,sh_group_size ) == 0 )  THEN
373!
374!--       If group boundary, start new IO group
375          node_grid(my_color)%nyn = local_dim_r(4,pe-1)
376          my_color = my_color + 1
377          group_start = pe
378          node_grid(my_color)%nxl = local_dim_r(1,group_start)
379          node_grid(my_color)%nxr = local_dim_r(2,group_start)
380          node_grid(my_color)%nys = local_dim_r(3,group_start)
381
382       ELSEIF ( local_dim_r(1,pe) /= node_grid(my_color)%nxl )  THEN
383!
384!--       If nxl changes, start new IO group
385          node_grid(my_color)%nyn = local_dim_r(4,pe-1)
386          my_color = my_color+1
387          group_start = pe
388          node_grid(my_color)%nxl = local_dim_r(1,group_start)
389          node_grid(my_color)%nxr = local_dim_r(2,group_start)
390          node_grid(my_color)%nys = local_dim_r(3,group_start)
391       ENDIF
392!
393!--    Save values for local PE
394       IF ( this%n_rank == pe )  THEN                                 !
395          color = my_color
396       ENDIF
397       IF ( n == this%n_npes-1 )  node_grid(my_color)%nyn = local_dim_r(4,pe)
398
399    ENDDO
400
401    IF ( this%n_rank == 0 )  THEN
402       color = 1
403    ENDIF
404
[4536]405    this%io_grid = node_grid(color)
406    this%io_grid%nnx = this%io_grid%nxr - this%io_grid%nxl + 1
407    this%io_grid%nny = this%io_grid%nyn - this%io_grid%nys + 1
408
[4534]409 END SUBROUTINE compute_color
410#endif
411
412 END SUBROUTINE sm_init_comm
413
414
[4617]415!
416!-- TODO: short description required, about the meaning of the following routine
417!--       part must be renamed particles!
418 SUBROUTINE sm_init_part( this )
[4534]419
[4617]420    IMPLICIT NONE
421
422    CLASS(sm_class), INTENT(INOUT) ::  this  !< pointer to access internal variables of this call
423
424#if defined( __parallel )
425    INTEGER(iwp) ::  color             !<
426    INTEGER(iwp) ::  ierr              !<
427    INTEGER(iwp) ::  max_n_npes        !< maximum number of PEs/node
[4620]428#endif
[4617]429
430    LOGICAL :: sm_active  !<
431
432
433    sm_active       = .TRUE.   ! particle IO always uses shared memory
434    this%comm_model = comm2d
435
436    this%no_shared_memory_in_this_run = .NOT. sm_active
437    this%comm_io = this%comm_model  ! preset in case of non shared-memory-IO
438
439    IF ( this%no_shared_memory_in_this_run )  THEN
440       this%iam_io_pe = .TRUE.
441       RETURN
442    ENDIF
443
444#if defined( __parallel )
445!
446!-- Determine, how many MPI threads are running on a node
447    this%iam_io_pe = .FALSE.
448    CALL MPI_COMM_SPLIT_TYPE( this%comm_model, MPI_COMM_TYPE_SHARED, 0, MPI_INFO_NULL,             &
449                              this%comm_node, ierr )
450    CALL MPI_COMM_SIZE( this%comm_node, this%n_npes, ierr )
451    CALL MPI_COMM_RANK( this%comm_node, this%n_rank, ierr )
452
453    CALL MPI_ALLREDUCE( this%n_npes, max_n_npes, 1, MPI_INTEGER, MPI_MAX, this%comm_model, ierr )
454
455!
456!-- TODO: better explanation
457!-- It has to be testet, if using memory blocks for an IO process (MPI shared Memory), or if it is
458!-- even better to use the complete node for MPI shared memory (this%nr_io_pe_per_node = 1).
459!-  In the latter case, the access to the MPI shared memory buffer is slower, the number of
460!-- particles to move between threads will be much smaller.
461    IF ( max_n_npes > 64 )  THEN
462!
463!--    Special configuration on the HLRN-IV system with 4 shared memory blocks/node
464       this%nr_io_pe_per_node = 4
465    ENDIF
466
467    IF ( this%nr_io_pe_per_node == 1 )  THEN
468!
469!--    This branch is not realized so far
470       this%iam_io_pe   = ( this%n_rank == 0 )
471       this%comm_shared = this%comm_node
472       CALL MPI_COMM_SIZE( this%comm_shared, this%sh_npes, ierr )
473       CALL MPI_COMM_RANK( this%comm_shared, this%sh_rank, ierr )
474
475    ELSEIF( this%nr_io_pe_per_node == 2 )  THEN
476
477       this%iam_io_pe = ( this%n_rank == 0  .OR.  this%n_rank == this%n_npes/2 )
478       IF ( this%n_rank < this%n_npes/2 )  THEN
479          color = 1
480       ELSE
481          color = 2
482       ENDIF
483       CALL MPI_COMM_SPLIT( this%comm_node, color, 0, this%comm_shared, ierr )
484       CALL MPI_COMM_SIZE( this%comm_shared, this%sh_npes, ierr )
485       CALL MPI_COMM_RANK( this%comm_shared, this%sh_rank, ierr )
486
487    ELSEIF( this%nr_io_pe_per_node == 4 )  THEN
488
489       this%iam_io_pe = ( this%n_rank == 0  .OR.  this%n_rank == this%n_npes/4  .OR.               &
490                          this%n_rank == this%n_npes/2  .OR.  this%n_rank == (3*this%n_npes)/4 )
491       IF ( this%n_rank < this%n_npes/4 )  THEN
492          color = 1
493       ELSEIF( this%n_rank < this%n_npes/2 )  THEN
494          color = 2
495       ELSEIF( this%n_rank < (3*this%n_npes)/4 )  THEN
496          color = 3
497       ELSE
498          color = 4
499       ENDIF
500       CALL MPI_COMM_SPLIT( this%comm_node, color, 0, this%comm_shared, ierr )
501       CALL MPI_COMM_SIZE( this%comm_shared, this%sh_npes, ierr )
502       CALL MPI_COMM_RANK( this%comm_shared, this%sh_rank, ierr )
503
504    ELSE
505
506       WRITE( *, * ) 'shared_memory_io_mod: internal error'
507       WRITE( *, * ) 'only 2 or 4 shared memory groups per node are allowed'
508       STOP
509
510    ENDIF
511
512!
513!-- Setup the shared memory area
514    CALL MPI_COMM_SPLIT( this%comm_node, color, 0, this%comm_shared, ierr )
515    CALL MPI_COMM_SIZE( this%comm_shared, this%sh_npes, ierr )
516    CALL MPI_COMM_RANK( this%comm_shared, this%sh_rank, ierr )
517
518!
519!-- Setup the communicator across the nodes depending on the shared memory rank.
520!-- All threads with shared memory rank 0 will be I/O threads.
521    color = this%sh_rank
522    CALL MPI_COMM_SPLIT( this%comm_model, color, 0, this%comm_io, ierr )
523
524    IF ( this%comm_io /= MPI_COMM_NULL )  THEN
525       CALL MPI_COMM_SIZE( this%comm_io, this%io_npes, ierr )
526       CALL MPI_COMM_RANK( this%comm_io, this%io_rank, ierr )
527    ELSE
528       this%io_npes = -1
529       this%io_rank = -1
530    ENDIF
531
532    IF ( this%sh_rank == 0 )  THEN
533       this%iam_io_pe = .TRUE.
534       this%io_pe_global_rank = myid
535    ENDIF
536    CALL MPI_BCAST( this%io_pe_global_rank, 1, MPI_INTEGER, 0, this%comm_shared, ierr )
537
538#else
539    this%iam_io_pe = .FALSE.
540#endif
541
542!    write(9,'(a,8i7)') 'sm_init_comm_part ',this%sh_rank,this%sh_npes,this%io_rank,this%io_npes
543
544 END SUBROUTINE sm_init_part
545
[4534]546!--------------------------------------------------------------------------------------------------!
547! Description:
548! ------------
549!> Function to return if shared Memory IO is active.
550!--------------------------------------------------------------------------------------------------!
[4591]551 FUNCTION is_sm_active( this ) RESULT( ac )
[4534]552
553    IMPLICIT NONE
554
[4591]555    CLASS(sm_class), INTENT(inout) ::  this  !<
[4534]556
[4591]557    LOGICAL ::  ac  !<
[4534]558
559    ac = .NOT. this%no_shared_memory_in_this_run
560
561 END FUNCTION is_sm_active
562
563
564#if defined( __parallel )
[4617]565
[4534]566!--------------------------------------------------------------------------------------------------!
567! Description:
568! ------------
[4617]569!> Allocate shared 1d-REAL (64 Bit) array on ALL threads
[4534]570!--------------------------------------------------------------------------------------------------!
[4617]571 SUBROUTINE sm_allocate_shared_1d_64( this, p1, d1, d2, win )
[4534]572
573    IMPLICIT NONE
574
[4617]575    CLASS(sm_class), INTENT(inout)  ::  this
[4534]576
[4617]577    INTEGER(iwp)                    ::  disp_unit
578    INTEGER(iwp), INTENT(IN)        ::  d1
579    INTEGER(iwp), INTENT(IN)        ::  d2
580    INTEGER(iwp), SAVE              ::  pe_from = 0
581    INTEGER(iwp), INTENT(OUT)       ::  win
[4534]582
[4617]583    INTEGER(KIND=MPI_ADDRESS_KIND)  ::  rem_size
584    INTEGER(KIND=MPI_ADDRESS_KIND)  ::  wsize
[4534]585
[4617]586    INTEGER, DIMENSION(1)           ::  buf_shape
[4534]587
[4617]588    REAL(dp), DIMENSION(:), POINTER ::  buf
589    REAL(dp), DIMENSION(:), POINTER ::  p1
[4534]590
[4617]591    TYPE(C_PTR), SAVE               ::  base_ptr
592    TYPE(C_PTR), SAVE               ::  rem_ptr
593
594
[4534]595    IF ( this%no_shared_memory_in_this_run )  RETURN
596!
597!-- Allocate shared memory on node rank 0 threads.
598    IF ( this%sh_rank == pe_from )  THEN
599       wsize = d2 - d1 + 1
600    ELSE
601       wsize = 1
602    ENDIF
[4617]603    wsize = wsize * dp  ! please note, size is always in bytes, independently of the displacement
604                        ! unit
605
606    CALL MPI_WIN_ALLOCATE_SHARED( wsize, dp, MPI_INFO_NULL, this%comm_shared,base_ptr, win, ierr )
607!
608!-- Get C-pointer of the memory located on node-rank pe_from (sh_rank == pe_from)
609    CALL MPI_WIN_SHARED_QUERY( win, pe_from, rem_size, disp_unit, rem_ptr, ierr )
610!
611!-- Convert C- to Fortran-pointer
612    buf_shape(1) = d2 - d1 + 1
613    CALL C_F_POINTER( rem_ptr, buf, buf_shape )
614    p1(d1:) => buf
615!
616!-- Allocate shared memory in round robin on all PEs of a node.
617    pe_from = MOD( pe_from, this%sh_npes )
618
619 END SUBROUTINE sm_allocate_shared_1d_64
620
621
622!--------------------------------------------------------------------------------------------------!
623! Description:
624! ------------
625!> Allocate shared 1d-REAL (32 Bit) array on ALL threads
626!--------------------------------------------------------------------------------------------------!
627 SUBROUTINE sm_allocate_shared_1d_32( this, p1, d1, d2, win )
628
629    IMPLICIT NONE
630
631    CLASS(sm_class), INTENT(inout)  ::  this
632
633    INTEGER(iwp)                    ::  disp_unit
634    INTEGER(iwp), INTENT(IN)        ::  d1
635    INTEGER(iwp), INTENT(IN)        ::  d2
636    INTEGER(iwp), SAVE              ::  pe_from = 0
637    INTEGER(iwp), INTENT(OUT)       ::  win
638
639    INTEGER(KIND=MPI_ADDRESS_KIND)  ::  rem_size
640    INTEGER(KIND=MPI_ADDRESS_KIND)  ::  wsize
641
642    INTEGER, DIMENSION(1)           ::  buf_shape
643
644    REAL(sp), DIMENSION(:), POINTER ::  buf
645    REAL(sp), DIMENSION(:), POINTER ::  p1
646
647    TYPE(C_PTR), SAVE               ::  base_ptr
648    TYPE(C_PTR), SAVE               ::  rem_ptr
649
650
651    IF ( this%no_shared_memory_in_this_run )  RETURN
652!
653!-- Allocate shared memory on node rank 0 threads.
654    IF ( this%sh_rank == pe_from )  THEN
655       wsize = d2 - d1 + 1
656    ELSE
657       wsize = 1
658    ENDIF
659    wsize = wsize * sp  ! Please note, size is always in bytes, independently of the displacement
[4534]660                       ! unit
661
[4617]662    CALL MPI_WIN_ALLOCATE_SHARED( wsize, sp, MPI_INFO_NULL, this%comm_shared,base_ptr, win, ierr )
[4534]663!
664!-- Get C-pointer of the memory located on node-rank pe_from (sh_rank == pe_from)
665    CALL MPI_WIN_SHARED_QUERY( win, pe_from, rem_size, disp_unit, rem_ptr, ierr )
666!
667!-- Convert C- to Fortran-pointer
668    buf_shape(1) = d2 - d1 + 1
669    CALL C_F_POINTER( rem_ptr, buf, buf_shape )
670    p1(d1:) => buf
671!
672!-- Allocate shared memory in round robin on all PEs of a node.
673    pe_from = MOD( pe_from, this%sh_npes )
674
[4617]675 END SUBROUTINE sm_allocate_shared_1d_32
[4534]676
677
678!--------------------------------------------------------------------------------------------------!
679! Description:
680! ------------
[4617]681!> Allocate shared 1d-INTEGER array on ALL threads
[4534]682!--------------------------------------------------------------------------------------------------!
[4617]683 SUBROUTINE sm_allocate_shared_1di( this, p1, d1, d2, win )
[4534]684
685    IMPLICIT NONE
686
[4617]687    CLASS(sm_class), INTENT(inout)  ::  this
[4534]688
[4617]689    INTEGER(iwp)                    ::  disp_unit
690    INTEGER(iwp), INTENT(IN)        ::  d1
691    INTEGER(iwp), INTENT(IN)        ::  d2
692    INTEGER(iwp), SAVE              ::  pe_from = 0
693    INTEGER(iwp), INTENT(OUT)       ::  win
[4534]694
[4617]695    INTEGER(KIND=MPI_ADDRESS_KIND)  ::  rem_size
696    INTEGER(KIND=MPI_ADDRESS_KIND)  ::  wsize
[4534]697
[4617]698    INTEGER, DIMENSION(1)           ::  buf_shape
[4534]699
[4617]700    INTEGER(iwp), DIMENSION(:), POINTER ::  buf
701    INTEGER(iwp), DIMENSION(:), POINTER ::  p1
[4534]702
[4617]703    TYPE(C_PTR), SAVE                   ::  base_ptr
704    TYPE(C_PTR), SAVE                   ::  rem_ptr
[4534]705
[4617]706
[4534]707    IF ( this%no_shared_memory_in_this_run )  RETURN
708!
709!-- Allocate shared memory on node rank 0 threads.
710    IF ( this%sh_rank == pe_from )  THEN
[4617]711       wsize = d2 - d1 + 1
712    ELSE
713       wsize = 1
714    ENDIF
715    wsize = wsize * iwp  ! Please note, size is always in bytes, independently of the displacement
716                       ! unit
717
718    CALL MPI_WIN_ALLOCATE_SHARED( wsize, iwp, MPI_INFO_NULL, this%comm_shared,base_ptr, win, ierr )
719!
720!-- Get C-pointer of the memory located on node-rank pe_from (sh_rank == pe_from)
721    CALL MPI_WIN_SHARED_QUERY( win, pe_from, rem_size, disp_unit, rem_ptr, ierr )
722!
723!-- Convert C- to Fortran-pointer
724    buf_shape(1) = d2 - d1 + 1
725    CALL C_F_POINTER( rem_ptr, buf, buf_shape )
726    p1(d1:) => buf
727!
728!-- Allocate shared memory in round robin on all PEs of a node.
729    pe_from = MOD( pe_from, this%sh_npes )
730
731 END SUBROUTINE sm_allocate_shared_1di
732
733
734!--------------------------------------------------------------------------------------------------!
735! Description:
736! ------------
737!> Allocate shared 2d-REAL array on ALL threads (64 Bit)
738!--------------------------------------------------------------------------------------------------!
739 SUBROUTINE sm_allocate_shared_2d_64( this, p2, n_nxlg, n_nxrg, n_nysg, n_nyng, win )
740
741    IMPLICIT NONE
742
743    CLASS(sm_class), INTENT(INOUT)    ::  this
744
745    INTEGER(iwp)                      ::  disp_unit
746    INTEGER(iwp), INTENT(IN)          ::  n_nxlg
747    INTEGER(iwp), INTENT(IN)          ::  n_nxrg
748    INTEGER(iwp), INTENT(IN)          ::  n_nyng
749    INTEGER(iwp), INTENT(IN)          ::  n_nysg
750    INTEGER(iwp), SAVE                ::  pe_from = 0
751    INTEGER(iwp), INTENT(OUT)         ::  win
752
753    INTEGER(KIND=MPI_ADDRESS_KIND)    ::  rem_size
754    INTEGER(KIND=MPI_ADDRESS_KIND)    ::  wsize
755
756    INTEGER(iwp), DIMENSION(2)        ::  buf_shape
757
758    REAL(dp), DIMENSION(:,:), POINTER ::  buf
759    REAL(dp), DIMENSION(:,:), POINTER ::  p2
760
761    TYPE(C_PTR), SAVE                 ::  base_ptr
762    TYPE(C_PTR), SAVE                 ::  rem_ptr
763
764
765    IF ( this%no_shared_memory_in_this_run )  RETURN
766!
767!-- Allocate shared memory on node rank 0 threads.
768    IF ( this%sh_rank == pe_from )  THEN
[4534]769       wsize = ( n_nyng - n_nysg + 1 ) * ( n_nxrg - n_nxlg + 1 )
770    ELSE
771       wsize = 1
772    ENDIF
773
[4617]774    wsize = wsize * dp  ! Please note, size is always in bytes, independently of the displacement
775                        ! unit
[4534]776
777    CALL MPI_WIN_ALLOCATE_SHARED( wsize, 8, MPI_INFO_NULL, this%comm_shared, base_ptr, win, ierr )
778!
779!-- Get C-pointer of the memory located on node-rank pe_from (sh_rank == pe_from)
780    CALL MPI_WIN_SHARED_QUERY( win, pe_from, rem_size, disp_unit, rem_ptr, ierr )
781!
782!-- Convert C- to Fortran-pointer
783    buf_shape(2) = n_nyng - n_nysg + 1
784    buf_shape(1) = n_nxrg - n_nxlg + 1
785    CALL C_F_POINTER( rem_ptr, buf, buf_shape )
786    p2(n_nxlg:, n_nysg:) => buf
787!
788!-- Allocate shared memory in round robin on all PEs of a node.
789    pe_from = MOD( pe_from, this%sh_npes )
790
[4617]791 END SUBROUTINE sm_allocate_shared_2d_64
[4534]792
793
794!--------------------------------------------------------------------------------------------------!
795! Description:
796! ------------
[4617]797!> Allocate shared 2d-REAL (32 Bit) array on ALL threads
798!--------------------------------------------------------------------------------------------------!
799 SUBROUTINE sm_allocate_shared_2d_32( this, p2, n_nxlg, n_nxrg, n_nysg, n_nyng, win )
800
801    IMPLICIT NONE
802
803    CLASS(sm_class), INTENT(INOUT)    ::  this
804
805    INTEGER(iwp)                      ::  disp_unit
806    INTEGER(iwp), INTENT(IN)          ::  n_nxlg
807    INTEGER(iwp), INTENT(IN)          ::  n_nxrg
808    INTEGER(iwp), INTENT(IN)          ::  n_nyng
809    INTEGER(iwp), INTENT(IN)          ::  n_nysg
810    INTEGER(iwp), SAVE                ::  pe_from = 0
811    INTEGER(iwp), INTENT(OUT)         ::  win
812
813    INTEGER(KIND=MPI_ADDRESS_KIND)    ::  rem_size
814    INTEGER(KIND=MPI_ADDRESS_KIND)    ::  wsize
815
816    INTEGER(iwp), DIMENSION(2)        ::  buf_shape
817
818    REAL(sp), DIMENSION(:,:), POINTER ::  buf
819    REAL(sp), DIMENSION(:,:), POINTER ::  p2
820
821    TYPE(C_PTR), SAVE                 ::  base_ptr
822    TYPE(C_PTR), SAVE                 ::  rem_ptr
823
824
825    IF ( this%no_shared_memory_in_this_run )  RETURN
826!
827!-- Allocate shared memory on node rank 0 threads.
828    IF ( this%sh_rank == pe_from )  THEN
829       wsize = ( n_nyng - n_nysg + 1 ) * ( n_nxrg - n_nxlg + 1 )
830    ELSE
831       wsize = 1
832    ENDIF
833
834    wsize = wsize * sp  ! Please note, size is always in bytes, independently of the displacement
835                        ! unit
836
837    CALL MPI_WIN_ALLOCATE_SHARED( wsize, dp, MPI_INFO_NULL, this%comm_shared, base_ptr, win, ierr )
838!
839!-- Get C-pointer of the memory located on node-rank pe_from (sh_rank == pe_from)
840    CALL MPI_WIN_SHARED_QUERY( win, pe_from, rem_size, disp_unit, rem_ptr, ierr )
841!
842!-- Convert C- to Fortran-pointer
843    buf_shape(2) = n_nyng - n_nysg + 1
844    buf_shape(1) = n_nxrg - n_nxlg + 1
845    CALL C_F_POINTER( rem_ptr, buf, buf_shape )
846    p2(n_nxlg:, n_nysg:) => buf
847!
848!-- Allocate shared memory in round robin on all PEs of a node.
849    pe_from = MOD( pe_from, this%sh_npes )
850
851 END SUBROUTINE sm_allocate_shared_2d_32
852
853
854!--------------------------------------------------------------------------------------------------!
855! Description:
856! ------------
[4534]857!> Allocate shared 2d-INTEGER array on ALL threads
858!--------------------------------------------------------------------------------------------------!
859 SUBROUTINE sm_allocate_shared_2di( this, p2i, n_nxlg, n_nxrg, n_nysg, n_nyng, win )
860
861    IMPLICIT NONE
862
[4591]863    CLASS(sm_class), INTENT(inout)        ::  this         !<
[4534]864
[4591]865    INTEGER(iwp)                          ::  disp_unit    !<
866    INTEGER(iwp), INTENT(IN)              ::  n_nxlg       !<
867    INTEGER(iwp), INTENT(IN)              ::  n_nxrg       !<
868    INTEGER(iwp), INTENT(IN)              ::  n_nyng       !<
869    INTEGER(iwp), INTENT(IN)              ::  n_nysg       !<
870    INTEGER(iwp), SAVE                    ::  pe_from = 0  !<
[4617]871    INTEGER(iwp), INTENT(OUT)             ::  win          !<
872
[4591]873    INTEGER(kind=MPI_ADDRESS_KIND)        ::  rem_size     !<
874    INTEGER(kind=MPI_ADDRESS_KIND)        ::  wsize        !<
[4534]875
[4591]876    INTEGER(iwp), DIMENSION(2)            ::  buf_shape    !<
[4534]877
[4591]878    INTEGER(iwp), DIMENSION(:,:), POINTER ::  buf          !<
879    INTEGER(iwp), DIMENSION(:,:), POINTER ::  p2i          !<
[4534]880
[4617]881    TYPE(C_PTR), SAVE                     ::  base_ptr     !<
882    TYPE(C_PTR), SAVE                     ::  rem_ptr      !<
[4534]883
884
885    IF ( this%no_shared_memory_in_this_run )  RETURN
886!
887!-- Allocate shared memory on node rank 0 threads.
888    IF ( this%sh_rank == pe_from )  THEN
889       wsize = ( n_nyng - n_nysg + 1 ) * ( n_nxrg - n_nxlg + 1 )
890    ELSE
891       wsize = 1
892    ENDIF
893
894    wsize = wsize * 4  ! Please note, size is always in bytes, independently of the displacement
895                       ! unit
896
897    CALL MPI_WIN_ALLOCATE_SHARED( wsize, 4, MPI_INFO_NULL, this%comm_shared, base_ptr, win, ierr )
898!
899!-- Get C-pointer of the memory located on node-rank pe_from (sh_rank == pe_from)
900    CALL MPI_WIN_SHARED_QUERY( win, pe_from, rem_size, disp_unit, rem_ptr, ierr )
901!
902!-- Convert C- to Fortran-pointer
903    buf_shape(2) = n_nyng - n_nysg + 1
904    buf_shape(1) = n_nxrg - n_nxlg + 1
905    CALL C_F_POINTER( rem_ptr, buf, buf_shape )
906    p2i(n_nxlg:, n_nysg:) => buf
907!
908!-- Allocate shared memory in round robin on all PEs of a node.
909    pe_from = MOD( pe_from, this%sh_npes )
910
911 END SUBROUTINE sm_allocate_shared_2di
912
913
914!--------------------------------------------------------------------------------------------------!
915! Description:
916! ------------
[4617]917!> Allocate shared 3d-REAL (64 Bit) array on ALL threads
[4534]918!--------------------------------------------------------------------------------------------------!
[4617]919 SUBROUTINE sm_allocate_shared_3d_64( this, p3, d1s, d1e, d2s, d2e, d3s, d3e, win )
[4534]920
921    IMPLICIT NONE
922
[4591]923    CLASS(sm_class), INTENT(inout)      ::  this         !<
[4534]924
[4591]925    INTEGER                             ::  disp_unit    !<
926    INTEGER, INTENT(IN)                 ::  d1e          !<
927    INTEGER, INTENT(IN)                 ::  d1s          !<
928    INTEGER, INTENT(IN)                 ::  d2e          !<
929    INTEGER, INTENT(IN)                 ::  d2s          !<
930    INTEGER, INTENT(IN)                 ::  d3e          !<
931    INTEGER, INTENT(IN)                 ::  d3s          !<
932    INTEGER, SAVE                       ::  pe_from = 0  !<
[4617]933    INTEGER, INTENT(OUT)                ::  win          !<
934
[4591]935    INTEGER(KIND=MPI_ADDRESS_KIND)      ::  rem_size     !<
936    INTEGER(KIND=MPI_ADDRESS_KIND)      ::  wsize        !<
[4534]937
[4591]938    INTEGER, DIMENSION(3)               ::  buf_shape    !<
[4534]939
[4617]940    REAL(dp), DIMENSION(:,:,:), POINTER ::  buf          !<
941    REAL(dp), DIMENSION(:,:,:), POINTER ::  p3           !<
[4534]942
[4591]943    TYPE(C_PTR), SAVE                   ::  base_ptr     !<
944    TYPE(C_PTR), SAVE                   ::  rem_ptr      !<
[4534]945
946
947    IF ( this%no_shared_memory_in_this_run )  RETURN
948!
949!-- Allocate shared memory on node rank 0 threads.
950    IF ( this%sh_rank == pe_from )  THEN
951       wsize = ( d3e - d3s + 1 ) * ( d2e - d2s + 1 ) * ( d1e - d1s + 1 )
952    ELSE
953       wsize = 1
954    ENDIF
955
[4617]956    wsize = wsize * dp ! Please note, size is always in bytes, independently of the displacement
[4534]957                       ! unit
958
[4617]959    CALL MPI_WIN_ALLOCATE_SHARED( wsize, dp, MPI_INFO_NULL, this%comm_shared, base_ptr, win, ierr )
[4534]960!
961!-- Get C-pointer of the memory located on node-rank pe_from (sh_rank == pe_from)
962    CALL MPI_WIN_SHARED_QUERY( win, pe_from, rem_size, disp_unit, rem_ptr, ierr )
963!
964!-- Convert C- to Fortran-pointer
965    buf_shape(3) = d3e - d3s + 1
966    buf_shape(2) = d2e - d2s + 1
967    buf_shape(1) = d1e - d1s + 1
968    CALL C_F_POINTER( rem_ptr, buf, buf_shape )
969    p3(d1s:,d2s:,d3s:) => buf
970!
971!-- Allocate shared memory in round robin on all PEs of a node.
972    pe_from = MOD( pe_from, this%sh_npes )
973
[4617]974 END SUBROUTINE sm_allocate_shared_3d_64
975
976
977!--------------------------------------------------------------------------------------------------!
978! Description:
979! ------------
980!> Allocate shared 3d-REAL (32 Bit) array on ALL threads
981!--------------------------------------------------------------------------------------------------!
982 SUBROUTINE sm_allocate_shared_3d_32( this, p3, d1s, d1e, d2s, d2e, d3s, d3e, win )
983
984    IMPLICIT NONE
985
986    CLASS(sm_class), INTENT(inout)      ::  this
987
988    INTEGER                             ::  disp_unit
989    INTEGER, INTENT(IN)                 ::  d1e
990    INTEGER, INTENT(IN)                 ::  d1s
991    INTEGER, INTENT(IN)                 ::  d2e
992    INTEGER, INTENT(IN)                 ::  d2s
993    INTEGER, INTENT(IN)                 ::  d3e
994    INTEGER, INTENT(IN)                 ::  d3s
995    INTEGER, SAVE                       ::  pe_from = 0
996    INTEGER, INTENT(OUT)                ::  win
997
998    INTEGER(KIND=MPI_ADDRESS_KIND)      ::  rem_size
999    INTEGER(KIND=MPI_ADDRESS_KIND)      ::  wsize
1000
1001    INTEGER, DIMENSION(3)               ::  buf_shape
1002
1003    REAL(sp), DIMENSION(:,:,:), POINTER ::  buf
1004    REAL(sp), DIMENSION(:,:,:), POINTER ::  p3
1005
1006    TYPE(C_PTR), SAVE                   ::  base_ptr
1007    TYPE(C_PTR), SAVE                   ::  rem_ptr
1008
1009
1010    IF ( this%no_shared_memory_in_this_run )  RETURN
1011!
1012!-- Allocate shared memory on node rank 0 threads.
1013    IF ( this%sh_rank == pe_from )  THEN
1014       wsize = ( d3e - d3s + 1 ) * ( d2e - d2s + 1 ) * ( d1e - d1s + 1 )
1015    ELSE
1016       wsize = 1
1017    ENDIF
1018
1019    wsize = wsize * sp ! Please note, size is always in bytes, independently of the displacement
1020                       ! unit
1021
1022    CALL MPI_WIN_ALLOCATE_SHARED( wsize, sp, MPI_INFO_NULL, this%comm_shared, base_ptr, win, ierr )
1023!
1024!-- Get C-pointer of the memory located on node-rank pe_from (sh_rank == pe_from)
1025    CALL MPI_WIN_SHARED_QUERY( win, pe_from, rem_size, disp_unit, rem_ptr, ierr )
1026!
1027!-- Convert C- to Fortran-pointer
1028    buf_shape(3) = d3e - d3s + 1
1029    buf_shape(2) = d2e - d2s + 1
1030    buf_shape(1) = d1e - d1s + 1
1031    CALL C_F_POINTER( rem_ptr, buf, buf_shape )
1032    p3(d1s:,d2s:,d3s:) => buf
1033!
1034!-- Allocate shared memory in round robin on all PEs of a node.
1035    pe_from = MOD( pe_from, this%sh_npes )
1036
1037 END SUBROUTINE sm_allocate_shared_3d_32
1038
[4628]1039
1040!--------------------------------------------------------------------------------------------------!
1041! Description:
1042! ------------
1043!> Allocate shared 3d-REAL (32 bit) array on ALL threads
1044!--------------------------------------------------------------------------------------------------!
1045 SUBROUTINE sm_allocate_shared_3di_32( this, p3, d1s, d1e, d2s, d2e, d3s, d3e, win )
1046
1047    IMPLICIT NONE
1048
1049    CLASS(sm_class), INTENT(inout)      ::  this
1050
1051    INTEGER                             ::  disp_unit
1052    INTEGER, INTENT(IN)                 ::  d1e
1053    INTEGER, INTENT(IN)                 ::  d1s
1054    INTEGER, INTENT(IN)                 ::  d2e
1055    INTEGER, INTENT(IN)                 ::  d2s
1056    INTEGER, INTENT(IN)                 ::  d3e
1057    INTEGER, INTENT(IN)                 ::  d3s
1058    INTEGER, SAVE                       ::  pe_from = 0
1059    INTEGER, INTENT(OUT)                ::  win
1060
1061    INTEGER(KIND=MPI_ADDRESS_KIND)      ::  rem_size
1062    INTEGER(KIND=MPI_ADDRESS_KIND)      ::  wsize
1063
1064    INTEGER, DIMENSION(3)               ::  buf_shape
1065
1066    INTEGER(isp), DIMENSION(:,:,:), POINTER ::  buf
1067    INTEGER(isp), DIMENSION(:,:,:), POINTER ::  p3
1068
1069    TYPE(C_PTR), SAVE                   ::  base_ptr
1070    TYPE(C_PTR), SAVE                   ::  rem_ptr
1071
1072
1073    IF ( this%no_shared_memory_in_this_run )  RETURN
1074!
1075!-- Allocate shared memory on node rank 0 threads.
1076    IF ( this%sh_rank == pe_from )  THEN
1077       wsize = ( d3e - d3s + 1 ) * ( d2e - d2s + 1 ) * ( d1e - d1s + 1 )
1078    ELSE
1079       wsize = 1
1080    ENDIF
1081
1082    wsize = wsize * isp ! Please note, size is always in bytes, independently of the displacement
1083                       ! unit
1084
1085    CALL MPI_WIN_ALLOCATE_SHARED( wsize, isp, MPI_INFO_NULL, this%comm_shared, base_ptr, win, ierr )
1086!
1087!-- Get C-pointer of the memory located on node-rank pe_from (sh_rank == pe_from)
1088    CALL MPI_WIN_SHARED_QUERY( win, pe_from, rem_size, disp_unit, rem_ptr, ierr )
1089!
1090!-- Convert C- to Fortran-pointer
1091    buf_shape(3) = d3e - d3s + 1
1092    buf_shape(2) = d2e - d2s + 1
1093    buf_shape(1) = d1e - d1s + 1
1094    CALL C_F_POINTER( rem_ptr, buf, buf_shape )
1095    p3(d1s:,d2s:,d3s:) => buf
1096!
1097!-- Allocate shared memory in round robin on all PEs of a node.
1098    pe_from = MOD( pe_from, this%sh_npes )
1099
1100 END SUBROUTINE sm_allocate_shared_3di_32
1101
1102
1103!--------------------------------------------------------------------------------------------------!
1104! Description:
1105! ------------
1106!> Allocate shared 3d-REAL (64 bit) array on ALL threads
1107!--------------------------------------------------------------------------------------------------!
1108 SUBROUTINE sm_allocate_shared_3di_64( this, p3, d1s, d1e, d2s, d2e, d3s, d3e, win )
1109
1110    IMPLICIT NONE
1111
1112    CLASS(sm_class), INTENT(inout)      ::  this         !<
1113
1114    INTEGER                             ::  disp_unit    !<
1115    INTEGER, INTENT(IN)                 ::  d1e          !<
1116    INTEGER, INTENT(IN)                 ::  d1s          !<
1117    INTEGER, INTENT(IN)                 ::  d2e          !<
1118    INTEGER, INTENT(IN)                 ::  d2s          !<
1119    INTEGER, INTENT(IN)                 ::  d3e          !<
1120    INTEGER, INTENT(IN)                 ::  d3s          !<
1121    INTEGER, SAVE                       ::  pe_from = 0  !<
1122    INTEGER, INTENT(OUT)                ::  win          !<
1123
1124    INTEGER(KIND=MPI_ADDRESS_KIND)      ::  rem_size     !<
1125    INTEGER(KIND=MPI_ADDRESS_KIND)      ::  wsize        !<
1126
1127    INTEGER, DIMENSION(3)               ::  buf_shape    !<
1128
1129    INTEGER(idp), DIMENSION(:,:,:), POINTER ::  buf          !<
1130    INTEGER(idp), DIMENSION(:,:,:), POINTER ::  p3           !<
1131
1132    TYPE(C_PTR), SAVE                   ::  base_ptr     !<
1133    TYPE(C_PTR), SAVE                   ::  rem_ptr      !<
1134
1135
1136    IF ( this%no_shared_memory_in_this_run )  RETURN
1137!
1138!-- Allocate shared memory on node rank 0 threads.
1139    IF ( this%sh_rank == pe_from )  THEN
1140       wsize = ( d3e - d3s + 1 ) * ( d2e - d2s + 1 ) * ( d1e - d1s + 1 )
1141    ELSE
1142       wsize = 1
1143    ENDIF
1144
1145    wsize = wsize * idp ! Please note, size is always in bytes, independently of the displacement
1146                        ! unit
1147
1148    CALL MPI_WIN_ALLOCATE_SHARED( wsize, idp, MPI_INFO_NULL, this%comm_shared, base_ptr, win, ierr )
1149!
1150!-- Get C-pointer of the memory located on node-rank pe_from (sh_rank == pe_from)
1151    CALL MPI_WIN_SHARED_QUERY( win, pe_from, rem_size, disp_unit, rem_ptr, ierr )
1152!
1153!-- Convert C- to Fortran-pointer
1154    buf_shape(3) = d3e - d3s + 1
1155    buf_shape(2) = d2e - d2s + 1
1156    buf_shape(1) = d1e - d1s + 1
1157    CALL C_F_POINTER( rem_ptr, buf, buf_shape )
1158    p3(d1s:,d2s:,d3s:) => buf
1159!
1160!-- Allocate shared memory in round robin on all PEs of a node.
1161    pe_from = MOD( pe_from, this%sh_npes )
1162
1163 END SUBROUTINE sm_allocate_shared_3di_64
1164
[4534]1165#endif
1166
1167
1168!--------------------------------------------------------------------------------------------------!
1169! Description:
1170! ------------
1171!> ???
1172!--------------------------------------------------------------------------------------------------!
1173 SUBROUTINE sm_adjust_outer_boundary( this )
1174
1175    IMPLICIT NONE
1176
[4591]1177    CLASS(sm_class), INTENT(inout) ::  this  !<
[4534]1178
1179
1180    IF ( this%no_shared_memory_in_this_run )  RETURN
1181
1182    IF ( this%io_grid%nxl == 0 )  THEN
1183       this%io_grid%nxl = this%io_grid%nxl - nbgp
1184       this%io_grid%nnx = this%io_grid%nnx + nbgp
1185    ENDIF
1186
1187    IF ( this%io_grid%nxr == nx  .OR.  npex == -1 )  THEN   ! npex == -1 if -D__parallel not set
1188       this%io_grid%nxr = this%io_grid%nxr + nbgp
1189       this%io_grid%nnx = this%io_grid%nnx + nbgp
1190    ENDIF
1191
1192    IF ( this%io_grid%nys == 0 )  THEN
1193       this%io_grid%nys = this%io_grid%nys - nbgp
1194       this%io_grid%nny = this%io_grid%nny + nbgp
1195    ENDIF
1196
1197    IF ( this%io_grid%nyn == ny .OR.  npey == -1 )  THEN   ! npey == -1 if -D__parallel not set
1198       this%io_grid%nyn = this%io_grid%nyn + nbgp
1199       this%io_grid%nny = this%io_grid%nny + nbgp
1200    ENDIF
1201
1202    this%io_grid%nxl = this%io_grid%nxl + nbgp
1203    this%io_grid%nxr = this%io_grid%nxr + nbgp
1204    this%io_grid%nys = this%io_grid%nys + nbgp
1205    this%io_grid%nyn = this%io_grid%nyn + nbgp
1206    this%io_grid%nnx = this%io_grid%nnx
1207    this%io_grid%nny = this%io_grid%nny
1208
1209 END SUBROUTINE sm_adjust_outer_boundary
1210
1211
1212!--------------------------------------------------------------------------------------------------!
1213! Description:
1214! ------------
1215!> Deallocate shared aray and free related window.
1216!--------------------------------------------------------------------------------------------------!
1217 SUBROUTINE sm_free_shared( this, win )
1218
1219    IMPLICIT NONE
1220
[4591]1221    CLASS(sm_class), INTENT(inout) ::  this  !<
[4534]1222
[4591]1223    INTEGER(iwp), INTENT(INOUT)    ::  win   !<
[4534]1224
[4617]1225    IF ( this%no_shared_memory_in_this_run )  RETURN
[4534]1226#if defined( __parallel )
1227    CALL MPI_WIN_FREE( win, ierr )
1228#endif
[4617]1229    win = -1
[4534]1230
1231 END SUBROUTINE sm_free_shared
1232
1233
1234!--------------------------------------------------------------------------------------------------!
1235! Description:
1236! ------------
1237!> ...
1238!--------------------------------------------------------------------------------------------------!
1239 SUBROUTINE sm_node_barrier( this )
1240
1241    IMPLICIT NONE
1242
[4591]1243    CLASS(sm_class), INTENT(inout) ::  this  !<
[4534]1244
1245
1246    IF ( this%no_shared_memory_in_this_run )  RETURN
1247
1248#if defined( __parallel )
1249    CALL MPI_BARRIER( this%comm_shared, ierr )
1250#endif
1251
1252 END SUBROUTINE sm_node_barrier
1253
[4617]1254
1255 SUBROUTINE save_grid_into_this_class( this )
1256
1257    IMPLICIT NONE
1258
1259    CLASS(domain_decomposition_grid_features), INTENT(inout) ::  this  !<
1260
1261       this%myid     = myid      !<
1262       this%nnx      = nnx       !<
1263       this%nny      = nny       !<
1264       this%nx       = nx        !<
1265       this%nxl      = nxl       !<
1266       this%nxr      = nxr       !<
1267       this%ny       = ny        !<
1268       this%nyn      = nyn       !<
1269       this%nys      = nys       !<
1270       this%numprocs = numprocs  !<
1271       this%comm2d   = comm2d    !<
1272
1273 END SUBROUTINE save_grid_into_this_class
1274
1275
1276 SUBROUTINE activate_grid_from_this_class( this )
1277
1278    IMPLICIT NONE
1279
1280    CLASS(domain_decomposition_grid_features), INTENT(inout) ::  this  !<
1281
1282       myid     = this%myid      !<
1283       nnx      = this%nnx       !<
1284       nny      = this%nny       !<
1285       nx       = this%nx        !<
1286       nxl      = this%nxl       !<
1287       nxr      = this%nxr       !<
1288       ny       = this%ny        !<
1289       nyn      = this%nyn       !<
1290       nys      = this%nys       !<
1291       numprocs = this%numprocs  !<
1292       comm2d   = this%comm2d    !<
1293
1294 END SUBROUTINE activate_grid_from_this_class
1295
[4534]1296 END MODULE shared_memory_io_mod
Note: See TracBrowser for help on using the repository browser.