Home

Context Navigation

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

Last change on this file since 2815 was 2809, checked in by schwenkel, 7 years ago
Bugfix for gfortran: Replace the function C_SIZEOF with STORAGE_SIZE
Property svn:keywords set to `Id`
File size: 34.8 KB

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

Note: See TracBrowser for help on using the repository browser.

Download in other formats:

| Impressum | ©Leibniz Universität Hannover |