source: palm/trunk/SOURCE/pmc_parent_mod.f90 @ 2345

Last change on this file since 2345 was 2101, checked in by suehring, 8 years ago

last commit documented

  • Property svn:keywords set to Id
File size: 30.5 KB
RevLine 
[1933]1 MODULE pmc_parent
[1762]2
[2000]3!------------------------------------------------------------------------------!
[1762]4! This file is part of PALM.
5!
[2000]6! PALM is free software: you can redistribute it and/or modify it under the
7! terms of the GNU General Public License as published by the Free Software
8! Foundation, either version 3 of the License, or (at your option) any later
9! version.
[1762]10!
11! PALM is distributed in the hope that it will be useful, but WITHOUT ANY
12! WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
13! A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
14!
15! You should have received a copy of the GNU General Public License along with
16! PALM. If not, see <http://www.gnu.org/licenses/>.
17!
[2101]18! Copyright 1997-2017 Leibniz Universitaet Hannover
[2000]19!------------------------------------------------------------------------------!
[1762]20!
21! Current revisions:
22! ------------------
[1834]23!
[2001]24!
[1762]25! Former revisions:
26! -----------------
27! $Id: pmc_parent_mod.f90 2101 2017-01-05 16:42:31Z Giersch $
28!
[2001]29! 2000 2016-08-20 18:09:15Z knoop
30! Forced header and separation lines into 80 columns
31!
[1939]32! 1938 2016-06-13 15:26:05Z hellstea
33! Minor clean up.
34!
[1933]35! 1901 2016-05-04 15:39:38Z raasch
36! Module renamed. Code clean up. The words server/client changed to parent/child.
37!
[1901]38! 1900 2016-05-04 15:27:53Z raasch
39! re-formatted to match PALM style
40!
[1851]41! 1850 2016-04-08 13:29:27Z maronga
42! Module renamed
43!
44!
[1834]45! 1833 2016-04-07 14:23:03Z raasch
46! gfortran requires pointer attributes for some array declarations,
47! long line wrapped
48!
[1809]49! 1808 2016-04-05 19:44:00Z raasch
50! MPI module used by default on all machines
51!
[1798]52! 1797 2016-03-21 16:50:28Z raasch
53! introduction of different datatransfer modes
54!
[1792]55! 1791 2016-03-11 10:41:25Z raasch
56! Debug write-statements commented out
57!
[1787]58! 1786 2016-03-08 05:49:27Z raasch
[1933]59! change in child-parent data transfer: parent now gets data from child
60! instead that child put's it to the parent
[1787]61!
[1780]62! 1779 2016-03-03 08:01:28Z raasch
63! kind=dp replaced by wp,
64! error messages removed or changed to PALM style, dim_order removed
65! array management changed from linked list to sequential loop
66!
[1767]67! 1766 2016-02-29 08:37:15Z raasch
68! modifications to allow for using PALM's pointer version
69! +new routine pmc_s_set_active_data_array
70!
[1765]71! 1764 2016-02-28 12:45:19Z raasch
72! cpp-statement added (nesting can only be used in parallel mode)
73!
[1763]74! 1762 2016-02-25 12:31:13Z hellstea
75! Initial revision by K. Ketelsen
[1762]76!
77! Description:
78! ------------
79!
[1933]80! Parent part of Palm Model Coupler
81!-------------------------------------------------------------------------------!
[1762]82
[1764]83#if defined( __parallel )
[1900]84    USE, INTRINSIC ::  ISO_C_BINDING
[1762]85
[1808]86#if defined( __mpifh )
87    INCLUDE "mpif.h"
88#else
[1764]89    USE MPI
90#endif
[1900]91    USE kinds
[1933]92    USE pmc_general,                                                            &
93        ONLY: arraydef, childdef, da_namedef, da_namelen, pedef,                &
[1900]94              pmc_g_setname, pmc_max_array, pmc_max_models, pmc_sort
[1762]95
[1933]96    USE pmc_handle_communicator,                                                &
97        ONLY: m_model_comm,m_model_rank,m_model_npes, m_to_child_comm,          &
98              m_world_rank, pmc_parent_for_child
[1900]99
[1933]100    USE pmc_mpi_wrapper,                                                        &
[1900]101        ONLY: pmc_alloc_mem, pmc_bcast, pmc_time
102
103   IMPLICIT NONE
104
[1762]105   PRIVATE
106   SAVE
107
[1933]108   TYPE childindexdef
[1900]109      INTEGER                              ::  nrpoints       !<
110      INTEGER, DIMENSION(:,:), ALLOCATABLE ::  index_list_2d  !<
[1933]111   END TYPE childindexdef
[1762]112
[1933]113   TYPE(childdef), DIMENSION(pmc_max_models)       ::  children     !<
114   TYPE(childindexdef), DIMENSION(pmc_max_models)  ::  indchildren  !<
[1762]115
[1900]116   INTEGER ::  next_array_in_list = 0  !<
[1779]117
[1762]118
[1933]119   PUBLIC pmc_parent_for_child
[1762]120
121
[1933]122   INTERFACE pmc_parentinit
123      MODULE PROCEDURE  pmc_parentinit
124   END INTERFACE pmc_parentinit
[1779]125
[1900]126    INTERFACE pmc_s_set_2d_index_list
127        MODULE PROCEDURE pmc_s_set_2d_index_list
128    END INTERFACE pmc_s_set_2d_index_list
[1762]129
[1900]130    INTERFACE pmc_s_clear_next_array_list
131        MODULE PROCEDURE pmc_s_clear_next_array_list
132    END INTERFACE pmc_s_clear_next_array_list
[1762]133
[1900]134    INTERFACE pmc_s_getnextarray
135        MODULE PROCEDURE pmc_s_getnextarray
136    END INTERFACE pmc_s_getnextarray
[1762]137
[1900]138    INTERFACE pmc_s_set_dataarray
139        MODULE PROCEDURE pmc_s_set_dataarray_2d
140        MODULE PROCEDURE pmc_s_set_dataarray_3d
141    END INTERFACE pmc_s_set_dataarray
[1762]142
[1900]143    INTERFACE pmc_s_setind_and_allocmem
144        MODULE PROCEDURE pmc_s_setind_and_allocmem
145    END INTERFACE pmc_s_setind_and_allocmem
[1762]146
[1900]147    INTERFACE pmc_s_fillbuffer
148        MODULE PROCEDURE pmc_s_fillbuffer
149    END INTERFACE pmc_s_fillbuffer
[1766]150
[1900]151    INTERFACE pmc_s_getdata_from_buffer
152        MODULE PROCEDURE pmc_s_getdata_from_buffer
153    END INTERFACE pmc_s_getdata_from_buffer
[1762]154
[1900]155    INTERFACE pmc_s_set_active_data_array
156        MODULE PROCEDURE pmc_s_set_active_data_array
157    END INTERFACE pmc_s_set_active_data_array
[1762]158
[1933]159    PUBLIC pmc_parentinit, pmc_s_clear_next_array_list, pmc_s_fillbuffer,       &
160           pmc_s_getdata_from_buffer, pmc_s_getnextarray,                       &
161           pmc_s_setind_and_allocmem, pmc_s_set_active_data_array,              &
[1900]162           pmc_s_set_dataarray, pmc_s_set_2d_index_list
[1762]163
[1900]164 CONTAINS
[1762]165
166
[1933]167 SUBROUTINE pmc_parentinit
[1762]168
[1900]169    IMPLICIT NONE
[1762]170
[1933]171    INTEGER ::  childid   !<
[1900]172    INTEGER ::  i         !<
173    INTEGER ::  j         !<
174    INTEGER ::  istat     !<
[1762]175
176
[1933]177    DO  i = 1, SIZE( pmc_parent_for_child )-1
[1762]178
[1933]179       childid = pmc_parent_for_child( i )
[1900]180
[1933]181       children(childid)%model_comm = m_model_comm
182       children(childid)%inter_comm = m_to_child_comm(childid)
183
[1779]184!
[1900]185!--    Get rank and size
[1933]186       CALL MPI_COMM_RANK( children(childid)%model_comm,                        &
187                           children(childid)%model_rank, istat )
188       CALL MPI_COMM_SIZE( children(childid)%model_comm,                        &
189                           children(childid)%model_npes, istat )
190       CALL MPI_COMM_REMOTE_SIZE( children(childid)%inter_comm,                 &
191                                  children(childid)%inter_npes, istat )
192
[1900]193!
194!--    Intra communicater is used for MPI_GET
[1933]195       CALL MPI_INTERCOMM_MERGE( children(childid)%inter_comm, .FALSE.,         &
196                                 children(childid)%intra_comm, istat )
197       CALL MPI_COMM_RANK( children(childid)%intra_comm,                        &
198                           children(childid)%intra_rank, istat )
[1762]199
[1933]200       ALLOCATE( children(childid)%pes(children(childid)%inter_npes))
201
[1900]202!
[1933]203!--    Allocate array of TYPE arraydef for all child PEs to store information
[1900]204!--    of the transfer array
[1933]205       DO  j = 1, children(childid)%inter_npes
206         ALLOCATE( children(childid)%pes(j)%array_list(pmc_max_array) )
[1900]207       ENDDO
[1762]208
[1933]209       CALL get_da_names_from_child (childid)
[1762]210
[1900]211    ENDDO
[1762]212
[1933]213 END SUBROUTINE pmc_parentinit
[1762]214
215
216
[1933]217 SUBROUTINE pmc_s_set_2d_index_list( childid, index_list )
[1762]218
[1900]219     IMPLICIT NONE
[1762]220
[1933]221     INTEGER, INTENT(IN)                    :: childid     !<
[1900]222     INTEGER, DIMENSION(:,:), INTENT(INOUT) :: index_list  !<
[1762]223
[1900]224     INTEGER ::  ian    !<
225     INTEGER ::  ic     !<
226     INTEGER ::  ie     !<
227     INTEGER ::  ip     !<
228     INTEGER ::  is     !<
229     INTEGER ::  istat  !<
230     INTEGER ::  n      !<
[1762]231
232
[1900]233     IF ( m_model_rank == 0 )  THEN
[1933]234
[1900]235!
[1933]236!--     Sort to ascending parent PE order
[1900]237        CALL pmc_sort( index_list, 6 )
[1762]238
[1900]239        is = 1
240        DO  ip = 0, m_model_npes-1
[1933]241
[1900]242!
[1933]243!--        Split into parent PEs
[1900]244           ie = is - 1
[1933]245
[1900]246!
247!--        There may be no entry for this PE
248           IF ( is <= SIZE( index_list,2 )  .AND.  ie >= 0 )  THEN
[1762]249
[1900]250              DO WHILE ( index_list(6,ie+1 ) == ip )
251                 ie = ie + 1
252                 IF ( ie == SIZE( index_list,2 ) )  EXIT
253              ENDDO
[1762]254
[1900]255              ian = ie - is + 1
[1762]256
[1900]257           ELSE
258              is  = -1
259              ie  = -2
260              ian =  0
261           ENDIF
[1933]262
[1900]263!
[1933]264!--        Send data to other parent PEs
[1900]265           IF ( ip == 0 )  THEN
[1933]266              indchildren(childid)%nrpoints = ian
[1900]267              IF ( ian > 0)  THEN
[1933]268                  ALLOCATE( indchildren(childid)%index_list_2d(6,ian) )
269                  indchildren(childid)%index_list_2d(:,1:ian) =                 &
[1900]270                                                             index_list(:,is:ie)
271              ENDIF
272           ELSE
[1933]273              CALL MPI_SEND( ian, 1, MPI_INTEGER, ip, 1000, m_model_comm,       &
[1900]274                             istat )
275              IF ( ian > 0)  THEN
[1933]276                  CALL MPI_SEND( index_list(1,is), 6*ian, MPI_INTEGER, ip,      &
[1900]277                                 1001, m_model_comm, istat )
278              ENDIF
279           ENDIF
280           is = ie + 1
[1762]281
[1900]282        ENDDO
[1762]283
[1900]284     ELSE
[1762]285
[1933]286        CALL MPI_RECV( indchildren(childid)%nrpoints, 1, MPI_INTEGER, 0, 1000,  &
[1900]287                       m_model_comm, MPI_STATUS_IGNORE, istat )
[1933]288        ian = indchildren(childid)%nrpoints
[1779]289
[1900]290        IF ( ian > 0 )  THEN
[1933]291           ALLOCATE( indchildren(childid)%index_list_2d(6,ian) )
292           CALL MPI_RECV( indchildren(childid)%index_list_2d, 6*ian,            &
293                          MPI_INTEGER, 0, 1001, m_model_comm,                   &
[1900]294                          MPI_STATUS_IGNORE, istat)
295        ENDIF
[1779]296
[1900]297     ENDIF
[1779]298
[1933]299     CALL set_pe_index_list( childid, children(childid),                        &
300                             indchildren(childid)%index_list_2d,                &
301                             indchildren(childid)%nrpoints )
[1762]302
[1900]303 END SUBROUTINE pmc_s_set_2d_index_list
[1779]304
305
306
[1900]307 SUBROUTINE pmc_s_clear_next_array_list
[1833]308
[1900]309    IMPLICIT NONE
[1762]310
[1900]311    next_array_in_list = 0
[1762]312
[1900]313 END SUBROUTINE pmc_s_clear_next_array_list
[1762]314
315
316
[1933]317 LOGICAL FUNCTION pmc_s_getnextarray( childid, myname )
318
[1900]319!
320!-- List handling is still required to get minimal interaction with
321!-- pmc_interface
322!-- TODO: what does "still" mean? Is there a chance to change this!
323    CHARACTER(LEN=*), INTENT(OUT) ::  myname    !<
[1933]324    INTEGER(iwp), INTENT(IN)      ::  childid   !<
[1779]325
[1900]326    TYPE(arraydef), POINTER :: ar
327    TYPE(pedef), POINTER    :: ape
[1779]328
[1900]329    next_array_in_list = next_array_in_list + 1
[1933]330
[1900]331!
[1933]332!-- Array names are the same on all children PEs, so take first PE to get the name
333    ape => children(childid)%pes(1)
[1833]334
[1900]335    IF ( next_array_in_list > ape%nr_arrays )  THEN
[1933]336
[1900]337!
338!--    All arrays are done
339       pmc_s_getnextarray = .FALSE.
340       RETURN
341    ENDIF
[1779]342
[1900]343    ar => ape%array_list(next_array_in_list)
344    myname = ar%name
[1933]345
[1900]346!
347!-- Return true if legal array
348!-- TODO: what does this comment mean? Can there be non-legal arrays??
349    pmc_s_getnextarray = .TRUE.
[1762]350
[1900]351 END FUNCTION pmc_s_getnextarray
[1762]352
353
354
[1933]355 SUBROUTINE pmc_s_set_dataarray_2d( childid, array, array_2 )
[1900]356
357    IMPLICIT NONE
358
[1933]359    INTEGER,INTENT(IN) ::  childid   !<
[1900]360
361    REAL(wp), INTENT(IN), DIMENSION(:,:), POINTER           ::  array    !<
362    REAL(wp), INTENT(IN), DIMENSION(:,:), POINTER, OPTIONAL ::  array_2  !<
363
364    INTEGER               ::  nrdims      !<
365    INTEGER, DIMENSION(4) ::  dims        !<
366    TYPE(C_PTR)           ::  array_adr   !<
367    TYPE(C_PTR)           ::  second_adr  !<
368
369
370    dims      = 1
371    nrdims    = 2
372    dims(1)   = SIZE( array,1 )
373    dims(2)   = SIZE( array,2 )
374    array_adr = C_LOC( array )
375
376    IF ( PRESENT( array_2 ) )  THEN
377       second_adr = C_LOC(array_2)
[1933]378       CALL pmc_s_setarray( childid, nrdims, dims, array_adr,                   &
[1900]379                            second_adr = second_adr)
380    ELSE
[1933]381       CALL pmc_s_setarray( childid, nrdims, dims, array_adr )
[1900]382    ENDIF
383
384 END SUBROUTINE pmc_s_set_dataarray_2d
385
386
387
[1933]388 SUBROUTINE pmc_s_set_dataarray_3d( childid, array, nz_cl, nz, array_2 )
[1900]389
390    IMPLICIT NONE
391
[1933]392    INTEGER, INTENT(IN) ::  childid   !<
[1900]393    INTEGER, INTENT(IN) ::  nz        !<
394    INTEGER, INTENT(IN) ::  nz_cl     !<
395
396    REAL(wp), INTENT(IN), DIMENSION(:,:,:), POINTER           ::  array    !<
397    REAL(wp), INTENT(IN), DIMENSION(:,:,:), POINTER, OPTIONAL ::  array_2  !<
398
399    INTEGER               ::  nrdims      !<
400    INTEGER, DIMENSION(4) ::  dims        !<
401    TYPE(C_PTR)           ::  array_adr   !<
402    TYPE(C_PTR)           ::  second_adr  !<
403
[1933]404!
[1900]405!-- TODO: the next assignment seems to be obsolete. Please check!
406    dims      = 1
407    dims      = 0
408    nrdims    = 3
409    dims(1)   = SIZE( array,1 )
410    dims(2)   = SIZE( array,2 )
411    dims(3)   = SIZE( array,3 )
412    dims(4)   = nz_cl+dims(1)-nz  ! works for first dimension 1:nz and 0:nz+1
413
414    array_adr = C_LOC(array)
415
[1766]416!
[1900]417!-- In PALM's pointer version, two indices have to be stored internally.
418!-- The active address of the data array is set in swap_timelevel.
419    IF ( PRESENT( array_2 ) )  THEN
420      second_adr = C_LOC( array_2 )
[1933]421      CALL pmc_s_setarray( childid, nrdims, dims, array_adr,                    &
[1900]422                           second_adr = second_adr)
423    ELSE
[1933]424       CALL pmc_s_setarray( childid, nrdims, dims, array_adr )
[1900]425    ENDIF
[1762]426
[1900]427 END SUBROUTINE pmc_s_set_dataarray_3d
[1762]428
[1779]429
430
[1933]431 SUBROUTINE pmc_s_setind_and_allocmem( childid )
[1779]432
[1933]433    USE control_parameters,                                                     &
[1900]434        ONLY:  message_string
435
436    IMPLICIT NONE
437
[1786]438!
[1933]439!-- Naming convention for appendices:   _pc  -> parent to child transfer
440!--                                     _cp  -> child to parent transfer
441!--                                     send -> parent to child transfer
442!--                                     recv -> child to parent transfer
443    INTEGER, INTENT(IN) ::  childid   !<
[1762]444
[1900]445    INTEGER                        ::  arlen    !<
446    INTEGER                        ::  i        !<
447    INTEGER                        ::  ierr     !<
448    INTEGER                        ::  istat    !<
449    INTEGER                        ::  j        !<
450    INTEGER                        ::  myindex  !<
451    INTEGER                        ::  rcount   !< count MPI requests
452    INTEGER                        ::  tag      !<
[1762]453
[1900]454    INTEGER(idp)                   ::  bufsize  !< size of MPI data window
455    INTEGER(KIND=MPI_ADDRESS_KIND) ::  winsize  !<
456
457    INTEGER, DIMENSION(1024)       ::  req      !<
458
459    TYPE(C_PTR)             ::  base_ptr  !<
460    TYPE(pedef), POINTER    ::  ape       !<
461    TYPE(arraydef), POINTER ::  ar        !<
462
[1933]463    REAL(wp),DIMENSION(:), POINTER, SAVE ::  base_array_pc  !< base array for parent to child transfer
464    REAL(wp),DIMENSION(:), POINTER, SAVE ::  base_array_cp  !< base array for child to parent transfer
[1900]465
[1786]466!
[1933]467!-- Parent to child direction
[1900]468    myindex = 1
469    rcount  = 0
470    bufsize = 8
[1762]471
[1786]472!
[1900]473!-- First stride: compute size and set index
[1933]474    DO  i = 1, children(childid)%inter_npes
[1762]475
[1933]476       ape => children(childid)%pes(i)
[1900]477       tag = 200
[1762]478
[1900]479       DO  j = 1, ape%nr_arrays
[1762]480
[1900]481          ar  => ape%array_list(j)
482          IF ( ar%nrdims == 2 )  THEN
483             arlen = ape%nrele
484          ELSEIF ( ar%nrdims == 3 )  THEN
485             arlen = ape%nrele * ar%a_dim(4)
486          ELSE
487             arlen = -1
488          ENDIF
489          ar%sendindex = myindex
[1762]490
[1900]491          tag    = tag + 1
492          rcount = rcount + 1
[1933]493          CALL MPI_ISEND( myindex, 1, MPI_INTEGER, i-1, tag,                    &
494                          children(childid)%inter_comm, req(rcount), ierr )
495
[1900]496!
497!--       Maximum of 1024 outstanding requests
[1933]498!--       TODO: what does this limit mean?
[1900]499          IF ( rcount == 1024 )  THEN
500             CALL MPI_WAITALL( rcount, req, MPI_STATUSES_IGNORE, ierr )
501             rcount = 0
502          ENDIF
[1762]503
[1900]504          myindex = myindex + arlen
505          bufsize = bufsize + arlen
506          ar%sendsize = arlen
507
508       ENDDO
509
510       IF ( rcount > 0 )  THEN
511          CALL MPI_WAITALL( rcount, req, MPI_STATUSES_IGNORE, ierr )
512       ENDIF
513
514    ENDDO
515
[1786]516!
[1933]517!-- Create RMA (One Sided Communication) window for data buffer parent to
518!-- child transfer.
[1900]519!-- The buffer of MPI_GET (counterpart of transfer) can be PE-local, i.e.
520!-- it can but must not be part of the MPI RMA window. Only one RMA window is
521!-- required to prepare the data for
[1933]522!--                       parent -> child transfer on the parent side
[1900]523!-- and for
[1933]524!--                       child -> parent transfer on the child side
525    CALL pmc_alloc_mem( base_array_pc, bufsize )
526    children(childid)%totalbuffersize = bufsize * wp
[1762]527
[1900]528    winsize = bufsize * wp
[1933]529    CALL MPI_WIN_CREATE( base_array_pc, winsize, wp, MPI_INFO_NULL,             &
530                         children(childid)%intra_comm,                          &
531                         children(childid)%win_parent_child, ierr )
532
[1786]533!
[1900]534!-- Open window to set data
[1933]535    CALL MPI_WIN_FENCE( 0, children(childid)%win_parent_child, ierr )
536
[1900]537!
538!-- Second stride: set buffer pointer
[1933]539    DO  i = 1, children(childid)%inter_npes
[1786]540
[1933]541       ape => children(childid)%pes(i)
[1786]542
[1900]543       DO  j = 1, ape%nr_arrays
[1786]544
[1900]545          ar => ape%array_list(j)
[1933]546          ar%sendbuf = C_LOC( base_array_pc(ar%sendindex) )
[1786]547
[1933]548          IF ( ar%sendindex + ar%sendsize > bufsize )  THEN             
549             WRITE( message_string, '(a,i4,4i7,1x,a)' )                         &
[1938]550                    'parent buffer too small ',i,                               &
[1933]551                    ar%sendindex,ar%sendsize,ar%sendindex+ar%sendsize,          &
552                    bufsize,trim(ar%name)
553             CALL message( 'pmc_s_setind_and_allocmem', 'PA0429', 3, 2, 0, 6, 0 )
[1900]554          ENDIF
555       ENDDO
556    ENDDO
[1786]557
[1900]558!
[1933]559!-- Child to parent direction
[1900]560    bufsize = 8
[1933]561
[1900]562!
563!-- First stride: compute size and set index
[1933]564    DO  i = 1, children(childid)%inter_npes
[1786]565
[1933]566       ape => children(childid)%pes(i)
[1900]567       tag = 300
[1786]568
[1900]569       DO  j = 1, ape%nr_arrays
[1786]570
[1900]571          ar => ape%array_list(j)
[1933]572
[1900]573!
[1933]574!--       Receive index from child
[1900]575          tag = tag + 1
[1933]576          CALL MPI_RECV( myindex, 1, MPI_INTEGER, i-1, tag,                     &
577                         children(childid)%inter_comm, MPI_STATUS_IGNORE, ierr )
[1786]578
[1900]579          IF ( ar%nrdims == 3 )  THEN
580             bufsize = MAX( bufsize, ape%nrele * ar%a_dim(4) )
581          ELSE
582             bufsize = MAX( bufsize, ape%nrele )
583          ENDIF
584          ar%recvindex = myindex
[1786]585
[1900]586        ENDDO
[1762]587
[1900]588    ENDDO
[1762]589
[1900]590!
591!-- Create RMA (one sided communication) data buffer.
592!-- The buffer for MPI_GET can be PE local, i.e. it can but must not be part of
593!-- the MPI RMA window
[1933]594    CALL pmc_alloc_mem( base_array_cp, bufsize, base_ptr )
595    children(childid)%totalbuffersize = bufsize * wp
[1762]596
[1933]597    CALL MPI_BARRIER( children(childid)%intra_comm, ierr )
598
[1900]599!
600!-- Second stride: set buffer pointer
[1933]601    DO  i = 1, children(childid)%inter_npes
[1786]602
[1933]603       ape => children(childid)%pes(i)
[1762]604
[1900]605       DO  j = 1, ape%nr_arrays
606          ar => ape%array_list(j)
607          ar%recvbuf = base_ptr
608       ENDDO
[1762]609
[1900]610    ENDDO
[1762]611
[1900]612 END SUBROUTINE pmc_s_setind_and_allocmem
[1762]613
614
[1797]615
[1933]616 SUBROUTINE pmc_s_fillbuffer( childid, waittime )
[1762]617
[1900]618    IMPLICIT NONE
[1762]619
[1933]620    INTEGER, INTENT(IN)             ::  childid   !<
[1762]621
[1900]622    REAL(wp), INTENT(OUT), OPTIONAL ::  waittime  !<
[1779]623
[1900]624    INTEGER               ::  ierr     !<
625    INTEGER               ::  ij       !<
626    INTEGER               ::  ip       !<
627    INTEGER               ::  istat    !<
628    INTEGER               ::  j        !<
629    INTEGER               ::  myindex  !<
[1762]630
[1900]631    INTEGER, DIMENSION(1) ::  buf_shape
[1779]632
[1900]633    REAL(wp)                            ::  t1       !<
634    REAL(wp)                            ::  t2       !<
635    REAL(wp), POINTER, DIMENSION(:)     ::  buf      !<
636    REAL(wp), POINTER, DIMENSION(:,:)   ::  data_2d  !<
637    REAL(wp), POINTER, DIMENSION(:,:,:) ::  data_3d  !<
[1762]638
[1900]639    TYPE(pedef), POINTER    ::  ape  !<
640    TYPE(arraydef), POINTER ::  ar   !<
[1762]641
[1900]642!
[1933]643!-- Synchronization of the model is done in pmci_synchronize.
644!-- Therefor the RMA window can be filled without
[1900]645!-- sychronization at this point and a barrier is not necessary.
646!-- Please note that waittime has to be set in pmc_s_fillbuffer AND
647!-- pmc_c_getbuffer
648    IF ( PRESENT( waittime) )  THEN
649      t1 = pmc_time()
[1933]650      CALL MPI_BARRIER( children(childid)%intra_comm, ierr )
[1900]651      t2 = pmc_time()
652      waittime = t2- t1
653    ENDIF
[1786]654
[1933]655    DO  ip = 1, children(childid)%inter_npes
[1786]656
[1933]657       ape => children(childid)%pes(ip)
[1786]658
[1900]659       DO  j = 1, ape%nr_arrays
[1786]660
[1900]661          ar => ape%array_list(j)
662          myindex = 1
663
664          IF ( ar%nrdims == 2 )  THEN
665
666             buf_shape(1) = ape%nrele
667             CALL C_F_POINTER( ar%sendbuf, buf, buf_shape )
668             CALL C_F_POINTER( ar%data, data_2d, ar%a_dim(1:2) )
669             DO  ij = 1, ape%nrele
670                buf(myindex) = data_2d(ape%locind(ij)%j,ape%locind(ij)%i)
671                myindex = myindex + 1
672             ENDDO
673
674          ELSEIF ( ar%nrdims == 3 )  THEN
675
676             buf_shape(1) = ape%nrele*ar%a_dim(4)
677             CALL C_F_POINTER( ar%sendbuf, buf, buf_shape )
678             CALL C_F_POINTER( ar%data, data_3d, ar%a_dim(1:3) )
679             DO  ij = 1, ape%nrele
[1933]680                buf(myindex:myindex+ar%a_dim(4)-1) =                            &
[1900]681                        data_3d(1:ar%a_dim(4),ape%locind(ij)%j,ape%locind(ij)%i)
682                myindex = myindex + ar%a_dim(4)
683             ENDDO
684
685          ENDIF
686
687        ENDDO
688
689    ENDDO
[1933]690
[1786]691!
[1900]692!-- Buffer is filled
[1933]693    CALL MPI_BARRIER( children(childid)%intra_comm, ierr )
[1786]694
[1900]695 END SUBROUTINE pmc_s_fillbuffer
[1786]696
[1762]697
698
[1933]699 SUBROUTINE pmc_s_getdata_from_buffer( childid, waittime )
[1762]700
[1900]701    IMPLICIT NONE
[1762]702
[1933]703    INTEGER, INTENT(IN)             ::  childid      !<
704    REAL(wp), INTENT(OUT), OPTIONAL ::  waittime     !<
[1762]705
[1933]706    INTEGER                        ::  ierr          !<
707    INTEGER                        ::  ij            !<
708    INTEGER                        ::  ip            !<
709    INTEGER                        ::  istat         !<
710    INTEGER                        ::  j             !<
711    INTEGER                        ::  myindex       !<
712    INTEGER                        ::  nr            !<
713    INTEGER                        ::  target_pe     !<
714    INTEGER(kind=MPI_ADDRESS_KIND) ::  target_disp   !<
[1762]715
[1933]716    INTEGER, DIMENSION(1)          ::  buf_shape     !<
[1762]717
[1900]718    REAL(wp)                            ::  t1       !<
719    REAL(wp)                            ::  t2       !<
720    REAL(wp), POINTER, DIMENSION(:)     ::  buf      !<
721    REAL(wp), POINTER, DIMENSION(:,:)   ::  data_2d  !<
722    REAL(wp), POINTER, DIMENSION(:,:,:) ::  data_3d  !<
[1762]723
[1900]724    TYPE(pedef), POINTER    ::  ape  !<
725    TYPE(arraydef), POINTER ::  ar   !<
[1762]726
727
[1900]728    t1 = pmc_time()
[1933]729
[1900]730!
[1933]731!-- Wait for child to fill buffer
732    CALL MPI_BARRIER( children(childid)%intra_comm, ierr )
[1900]733    t2 = pmc_time() - t1
734    IF ( PRESENT( waittime ) )  waittime = t2
[1933]735
[1900]736!
737!-- TODO: check next statement
738!-- Fence might do it, test later
[1933]739!-- CALL MPI_WIN_FENCE( 0, children(childid)%win_parent_child, ierr)
740    CALL MPI_BARRIER( children(childid)%intra_comm, ierr )
[1762]741
[1933]742    DO  ip = 1, children(childid)%inter_npes
[1762]743
[1933]744       ape => children(childid)%pes(ip)
[1762]745
[1900]746       DO  j = 1, ape%nr_arrays
[1762]747
[1900]748          ar => ape%array_list(j)
[1762]749
[1900]750          IF ( ar%recvindex < 0 )  CYCLE
[1766]751
[1900]752          IF ( ar%nrdims == 2 )  THEN
753             nr = ape%nrele
754          ELSEIF ( ar%nrdims == 3 )  THEN
755             nr = ape%nrele * ar%a_dim(4)
756          ENDIF
[1766]757
[1900]758          buf_shape(1) = nr
759          CALL C_F_POINTER( ar%recvbuf, buf, buf_shape )
[1766]760
[1900]761!
762!--       MPI passive target RMA
763          IF ( nr > 0 )  THEN
764             target_disp = ar%recvindex - 1
[1933]765
[1900]766!
[1933]767!--          Child PEs are located behind parent PEs
[1900]768             target_pe = ip - 1 + m_model_npes
[1933]769             CALL MPI_WIN_LOCK( MPI_LOCK_SHARED, target_pe, 0,                  &
770                                children(childid)%win_parent_child, ierr )
771             CALL MPI_GET( buf, nr, MPI_REAL, target_pe, target_disp, nr,       &
772                           MPI_REAL, children(childid)%win_parent_child, ierr )
773             CALL MPI_WIN_UNLOCK( target_pe,                                    &
774                                  children(childid)%win_parent_child, ierr )
[1900]775          ENDIF
[1766]776
[1900]777          myindex = 1
778          IF ( ar%nrdims == 2 )  THEN
[1766]779
[1900]780             CALL C_F_POINTER( ar%data, data_2d, ar%a_dim(1:2) )
781             DO  ij = 1, ape%nrele
782                data_2d(ape%locind(ij)%j,ape%locind(ij)%i) = buf(myindex)
783                myindex = myindex + 1
784             ENDDO
[1766]785
[1900]786          ELSEIF ( ar%nrdims == 3 )  THEN
[1762]787
[1900]788             CALL C_F_POINTER( ar%data, data_3d, ar%a_dim(1:3))
789             DO  ij = 1, ape%nrele
[1933]790                data_3d(1:ar%a_dim(4),ape%locind(ij)%j,ape%locind(ij)%i) =      &
[1900]791                                              buf(myindex:myindex+ar%a_dim(4)-1)
792                myindex = myindex + ar%a_dim(4)
793             ENDDO
[1762]794
[1900]795          ENDIF
[1762]796
[1900]797       ENDDO
[1762]798
[1900]799    ENDDO
[1762]800
[1900]801 END SUBROUTINE pmc_s_getdata_from_buffer
[1762]802
803
804
[1933]805 SUBROUTINE get_da_names_from_child( childid )
806
[1900]807!
[1933]808!-- Get data array description and name from child
[1900]809    IMPLICIT NONE
[1762]810
[1933]811    INTEGER, INTENT(IN) ::  childid  !<
[1762]812
[1900]813    TYPE(da_namedef) ::  myname  !<
[1762]814
[1900]815    DO
[1933]816       CALL pmc_bcast( myname%couple_index, 0, comm=m_to_child_comm(childid) )
[1900]817       IF ( myname%couple_index == -1 )  EXIT
[1933]818       CALL pmc_bcast( myname%parentdesc,   0, comm=m_to_child_comm(childid) )
819       CALL pmc_bcast( myname%nameonparent, 0, comm=m_to_child_comm(childid) )
820       CALL pmc_bcast( myname%childdesc,    0, comm=m_to_child_comm(childid) )
821       CALL pmc_bcast( myname%nameonchild,  0, comm=m_to_child_comm(childid) )
[1762]822
[1933]823       CALL pmc_g_setname( children(childid), myname%couple_index,              &
824                           myname%nameonparent )
[1900]825   ENDDO
[1762]826
[1933]827 END SUBROUTINE get_da_names_from_child
[1762]828
829
830
[1933]831 SUBROUTINE pmc_s_setarray(childid, nrdims, dims, array_adr, second_adr )
832
[1900]833!
[1933]834!-- Set array for child inter PE 0
[1900]835    IMPLICIT NONE
[1762]836
[1933]837    INTEGER, INTENT(IN)               ::  childid    !<
838    INTEGER, INTENT(IN)               ::  nrdims     !<
839    INTEGER, INTENT(IN), DIMENSION(:) ::  dims       !<
[1762]840
[1900]841    TYPE(C_PTR), INTENT(IN)           :: array_adr   !<
842    TYPE(C_PTR), INTENT(IN), OPTIONAL :: second_adr  !<
[1762]843
[1900]844    INTEGER ::  i  !< local counter
[1762]845
[1900]846    TYPE(pedef), POINTER    ::  ape  !<
847    TYPE(arraydef), POINTER ::  ar   !<
[1762]848
849
[1933]850    DO  i = 1, children(childid)%inter_npes
[1762]851
[1933]852       ape => children(childid)%pes(i)
[1900]853       ar  => ape%array_list(next_array_in_list)
854       ar%nrdims = nrdims
855       ar%a_dim  = dims
856       ar%data   = array_adr
857
858       IF ( PRESENT( second_adr ) )  THEN
859          ar%po_data(1) = array_adr
860          ar%po_data(2) = second_adr
861       ELSE
862          ar%po_data(1) = C_NULL_PTR
863          ar%po_data(2) = C_NULL_PTR
864       ENDIF
865
866    ENDDO
867
868 END SUBROUTINE pmc_s_setarray
869
870
871
[1933]872 SUBROUTINE pmc_s_set_active_data_array( childid, iactive )
[1900]873
874    IMPLICIT NONE
875
[1933]876    INTEGER, INTENT(IN) ::  childid   !<
[1900]877    INTEGER, INTENT(IN) ::  iactive   !<
878
879    INTEGER :: i   !<
880    INTEGER :: ip  !<
881    INTEGER :: j   !<
882
883    TYPE(pedef), POINTER    ::  ape  !<
884    TYPE(arraydef), POINTER ::  ar   !<
885
[1933]886    DO  ip = 1, children(childid)%inter_npes
[1900]887
[1933]888       ape => children(childid)%pes(ip)
[1900]889
890       DO  j = 1, ape%nr_arrays
891
892          ar => ape%array_list(j)
893          IF ( iactive == 1  .OR.  iactive == 2 )  THEN
894             ar%data = ar%po_data(iactive)
895          ENDIF
896
897       ENDDO
898
899    ENDDO
900
901 END SUBROUTINE pmc_s_set_active_data_array
902
903
904
[1933]905 SUBROUTINE set_pe_index_list( childid, mychild, index_list, nrp )
[1900]906
907    IMPLICIT NONE
908
[1933]909    INTEGER, INTENT(IN)                 ::  childid     !<
[1900]910    INTEGER, INTENT(IN), DIMENSION(:,:) ::  index_list  !<
911    INTEGER, INTENT(IN)                 ::  nrp         !<
912
[1933]913    TYPE(childdef), INTENT(INOUT)       ::  mychild     !<
[1900]914
915    INTEGER                                 :: i        !<
916    INTEGER                                 :: ierr     !<
917    INTEGER                                 :: ind      !<
918    INTEGER                                 :: indwin   !<
919    INTEGER                                 :: indwin2  !<
920    INTEGER                                 :: i2       !<
921    INTEGER                                 :: j        !<
922    INTEGER                                 :: rempe    !<
923    INTEGER(KIND=MPI_ADDRESS_KIND)          :: winsize  !<
924
[1933]925    INTEGER, DIMENSION(mychild%inter_npes)  :: remind   !<
[1900]926
927    INTEGER, DIMENSION(:), POINTER          :: remindw  !<
928    INTEGER, DIMENSION(:), POINTER          :: rldef    !<
929
930    TYPE(pedef), POINTER                    :: ape      !<
931
932!
[1933]933!-- First, count entries for every remote child PE
934    DO  i = 1, mychild%inter_npes
935       ape => mychild%pes(i)
[1900]936       ape%nrele = 0
937    ENDDO
[1933]938
[1900]939!
940!-- Loop over number of coarse grid cells
941    DO  j = 1, nrp
942       rempe = index_list(5,j) + 1   ! PE number on remote PE
[1933]943       ape => mychild%pes(rempe)
944       ape%nrele = ape%nrele + 1     ! Increment number of elements for this child PE
[1900]945    ENDDO
946
[1933]947    DO  i = 1, mychild%inter_npes
948       ape => mychild%pes(i)
[1900]949       ALLOCATE( ape%locind(ape%nrele) )
950    ENDDO
951
952    remind = 0
953
954!
955!-- Second, create lists
956!-- Loop over number of coarse grid cells
957    DO  j = 1, nrp
958       rempe = index_list(5,j) + 1
[1933]959       ape => mychild%pes(rempe)
[1900]960       remind(rempe)     = remind(rempe)+1
961       ind               = remind(rempe)
962       ape%locind(ind)%i = index_list(1,j)
963       ape%locind(ind)%j = index_list(2,j)
964    ENDDO
[1933]965
[1900]966!
[1933]967!-- Prepare number of elements for children PEs
968    CALL pmc_alloc_mem( rldef, mychild%inter_npes*2 )
969
[1900]970!
[1933]971!-- Number of child PEs * size of INTEGER (i just arbitrary INTEGER)
972    winsize = mychild%inter_npes*c_sizeof(i)*2
[1900]973
[1933]974    CALL MPI_WIN_CREATE( rldef, winsize, iwp, MPI_INFO_NULL,                    &
975                         mychild%intra_comm, indwin, ierr )
976
[1900]977!
978!-- Open window to set data
979    CALL MPI_WIN_FENCE( 0, indwin, ierr )
980
981    rldef(1) = 0            ! index on remote PE 0
982    rldef(2) = remind(1)    ! number of elements on remote PE 0
[1933]983
[1900]984!
985!-- Reserve buffer for index array
[1933]986    DO  i = 2, mychild%inter_npes
[1900]987       i2          = (i-1) * 2 + 1
988       rldef(i2)   = rldef(i2-2) + rldef(i2-1) * 2  ! index on remote PE
[1933]989       rldef(i2+1) = remind(i)                      ! number of elements on remote PE
[1900]990    ENDDO
[1933]991
[1900]992!
[1933]993!-- Close window to allow child to access data
[1900]994    CALL MPI_WIN_FENCE( 0, indwin, ierr )
[1933]995
[1900]996!
[1933]997!-- Child has retrieved data
[1900]998    CALL MPI_WIN_FENCE( 0, indwin, ierr )
999
[1933]1000    i2 = 2 * mychild%inter_npes - 1
[1900]1001    winsize = ( rldef(i2) + rldef(i2+1) ) * 2
[1933]1002
[1900]1003!
1004!-- Make sure, MPI_ALLOC_MEM works
1005    winsize = MAX( winsize, 1 )
1006
1007    CALL pmc_alloc_mem( remindw, INT( winsize ) )
1008
1009    CALL MPI_BARRIER( m_model_comm, ierr )
[1933]1010    CALL MPI_WIN_CREATE( remindw, winsize*c_sizeof(i), iwp, MPI_INFO_NULL,      &
1011                         mychild%intra_comm, indwin2, ierr )
[1900]1012!
1013!-- Open window to set data
1014    CALL MPI_WIN_FENCE( 0, indwin2, ierr )
[1933]1015
[1900]1016!
1017!-- Create the 2D index list
1018    DO  j = 1, nrp
1019       rempe = index_list(5,j) + 1    ! PE number on remote PE
[1933]1020       ape => mychild%pes(rempe)
[1900]1021       i2    = rempe * 2 - 1
1022       ind   = rldef(i2) + 1
1023       remindw(ind)   = index_list(3,j)
1024       remindw(ind+1) = index_list(4,j)
1025       rldef(i2)      = rldef(i2)+2
1026    ENDDO
[1933]1027
[1900]1028!
[1933]1029!-- All data are set
[1900]1030    CALL MPI_WIN_FENCE( 0, indwin2, ierr )
[1933]1031
[1900]1032!
1033!-- Don't know why, but this barrier is necessary before windows can be freed
1034!-- TODO: find out why this is required
[1933]1035    CALL MPI_BARRIER( mychild%intra_comm, ierr )
[1900]1036
1037    CALL MPI_WIN_FREE( indwin, ierr )
1038    CALL MPI_WIN_FREE( indwin2, ierr )
1039
[1933]1040!
[1900]1041!-- TODO: check if the following idea needs to be done
1042!-- Sollte funktionieren, Problem mit MPI implementation
1043!-- https://www.lrz.de/services/software/parallel/mpi/onesided
1044!-- CALL MPI_Free_mem (remindw, ierr)
1045
1046 END SUBROUTINE set_pe_index_list
1047
[1764]1048#endif
[1933]1049 END MODULE pmc_parent
Note: See TracBrowser for help on using the repository browser.