Home

Context Navigation

source: palm/trunk/SOURCE/pmc_server.f90 @ 1788

Last change on this file since 1788 was 1787, checked in by raasch, 9 years ago
last commit documented
Property svn:keywords set to `Id`
File size: 29.3 KB

Rev	Line
[1762]	1	MODULE pmc_server
	2
	3	!--------------------------------------------------------------------------------!
	4	! This file is part of PALM.
	5	!
	6	! PALM is free software: you can redistribute it and/or modify it under the terms
	7	! of the GNU General Public License as published by the Free Software Foundation,
	8	! either version 3 of the License, or (at your option) any later version.
	9	!
	10	! PALM is distributed in the hope that it will be useful, but WITHOUT ANY
	11	! WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
	12	! A PARTICULAR PURPOSE. See the GNU General Public License for more details.
	13	!
	14	! You should have received a copy of the GNU General Public License along with
	15	! PALM. If not, see <http://www.gnu.org/licenses/>.
	16	!
	17	! Copyright 1997-2015 Leibniz Universitaet Hannover
	18	!--------------------------------------------------------------------------------!
	19	!
	20	! Current revisions:
	21	! ------------------
	22	!
[1787]	23	!
[1762]	24	! Former revisions:
	25	! -----------------
	26	! $Id: pmc_server.f90 1787 2016-03-08 06:57:00Z maronga $
	27	!
[1787]	28	! 1786 2016-03-08 05:49:27Z raasch
	29	! change in client-server data transfer: server now gets data from client
	30	! instead that client put's it to the server
	31	!
[1780]	32	! 1779 2016-03-03 08:01:28Z raasch
	33	! kind=dp replaced by wp,
	34	! error messages removed or changed to PALM style, dim_order removed
	35	! array management changed from linked list to sequential loop
	36	!
[1767]	37	! 1766 2016-02-29 08:37:15Z raasch
	38	! modifications to allow for using PALM's pointer version
	39	! +new routine pmc_s_set_active_data_array
	40	!
[1765]	41	! 1764 2016-02-28 12:45:19Z raasch
	42	! cpp-statement added (nesting can only be used in parallel mode)
	43	!
[1763]	44	! 1762 2016-02-25 12:31:13Z hellstea
	45	! Initial revision by K. Ketelsen
[1762]	46	!
	47	! Description:
	48	! ------------
	49	!
	50	! Server part of Palm Model Coupler
	51	!------------------------------------------------------------------------------!
	52
[1764]	53	#if defined( __parallel )
[1762]	54	use, intrinsic :: iso_c_binding
	55
[1764]	56	#if defined( __lc )
	57	USE MPI
	58	#else
	59	INCLUDE "mpif.h"
	60	#endif
	61	USE kinds
[1762]	62	USE PMC_general, ONLY: ClientDef, PMC_MAX_MODELL,PMC_sort, DA_NameDef, DA_Desclen, DA_Namelen, &
[1779]	63	PMC_G_SetName, PeDef, ArrayDef, PMC_MAX_ARRAY
[1762]	64	USE PMC_handle_communicator, ONLY: m_model_comm,m_model_rank,m_model_npes, m_to_client_comm, &
	65	PMC_Server_for_Client, m_world_rank
	66	USE PMC_MPI_wrapper, ONLY: PMC_Send_to_Client, PMC_Recv_from_Client, PMC_Bcast, PMC_Inter_Bcast, &
	67	PMC_Alloc_mem, PMC_Time
	68
	69	IMPLICIT none
	70	PRIVATE
	71	SAVE
	72
	73	TYPE ClientIndexDef
	74	INTEGER :: NrPoints
	75	INTEGER,DIMENSION(:,:),allocatable :: index_list_2d
	76	END TYPE ClientIndexDef
	77
	78	TYPE(ClientDef),DIMENSION(PMC_MAX_MODELL) :: Clients
	79	TYPE(ClientIndexDef),DIMENSION(PMC_MAX_MODELL) :: indClients
	80
[1779]	81	INTEGER :: next_array_in_list = 0
	82
[1762]	83	PUBLIC PMC_Server_for_Client
	84
	85	INTERFACE PMC_ServerInit
	86	MODULE procedure PMC_ServerInit
	87	END INTERFACE PMC_ServerInit
	88
	89	INTERFACE PMC_S_Set_2D_index_list
	90	MODULE procedure PMC_S_Set_2D_index_list
	91	END INTERFACE PMC_S_Set_2D_index_list
	92
[1779]	93	INTERFACE PMC_S_clear_next_array_list
	94	MODULE procedure PMC_S_clear_next_array_list
	95	END INTERFACE PMC_S_clear_next_array_list
	96
[1762]	97	INTERFACE PMC_S_GetNextArray
	98	MODULE procedure PMC_S_GetNextArray
	99	END INTERFACE PMC_S_GetNextArray
	100
	101	INTERFACE PMC_S_Set_DataArray
	102	MODULE procedure PMC_S_Set_DataArray_2d
	103	MODULE procedure PMC_S_Set_DataArray_3d
	104	END INTERFACE PMC_S_Set_DataArray
	105
	106	INTERFACE PMC_S_setInd_and_AllocMem
	107	MODULE procedure PMC_S_setInd_and_AllocMem
	108	END INTERFACE PMC_S_setInd_and_AllocMem
	109
	110	INTERFACE PMC_S_FillBuffer
	111	MODULE procedure PMC_S_FillBuffer
	112	END INTERFACE PMC_S_FillBuffer
	113
	114	INTERFACE PMC_S_GetData_from_Buffer
	115	MODULE procedure PMC_S_GetData_from_Buffer
	116	END INTERFACE PMC_S_GetData_from_Buffer
	117
[1766]	118	INTERFACE PMC_S_Set_Active_data_array
	119	MODULE procedure PMC_S_Set_Active_data_array
	120	END INTERFACE PMC_S_Set_Active_data_array
	121
[1762]	122	! PUBLIC section
	123
	124	PUBLIC PMC_ServerInit, PMC_S_Set_2D_index_list, PMC_S_GetNextArray, PMC_S_Set_DataArray
[1766]	125	PUBLIC PMC_S_setInd_and_AllocMem, PMC_S_FillBuffer, PMC_S_GetData_from_Buffer, PMC_S_Set_Active_data_array
[1779]	126	PUBLIC PMC_S_clear_next_array_list
[1762]	127
	128	CONTAINS
	129
	130	SUBROUTINE PMC_ServerInit
	131	IMPLICIT none
	132	INTEGER :: i
	133	INTEGER :: j
	134	INTEGER :: ClientId
	135	INTEGER :: istat
	136
	137	do i=1,size(PMC_Server_for_Client)-1
	138	if(m_model_comm == 0) write(0,*) 'PMC_Server: Initialize client Id',PMC_Server_for_Client(i)
	139
	140	ClientId = PMC_Server_for_Client(i)
	141
	142	Clients(ClientId)%model_comm = m_model_comm
	143	Clients(ClientId)%inter_comm = m_to_client_comm(ClientId)
	144
	145	! Get rank and size
	146	CALL MPI_Comm_rank (Clients(ClientId)%model_comm, Clients(ClientId)%model_rank, istat);
	147	CALL MPI_Comm_size (Clients(ClientId)%model_comm, Clients(ClientId)%model_npes, istat);
	148	CALL MPI_Comm_remote_size (Clients(ClientId)%inter_comm, Clients(ClientId)%inter_npes, istat);
	149
	150	! Intra communicater is used for MPI_Get
	151	CALL MPI_Intercomm_merge (Clients(ClientId)%inter_comm, .false., Clients(ClientId)%intra_comm, istat);
	152	CALL MPI_Comm_rank (Clients(ClientId)%intra_comm, Clients(ClientId)%intra_rank, istat);
	153
	154	write(9,*) 'ClientId ',i,ClientId,m_world_rank, Clients(ClientId)%inter_npes
	155
	156	ALLOCATE (Clients(ClientId)%PEs(Clients(ClientId)%inter_npes))
[1779]	157	!
	158	!-- Allocate for all client PEs an array of TYPE ArrayDef to store information of transfer array
	159	do j=1,Clients(ClientId)%inter_npes
	160	Allocate(Clients(ClientId)%PEs(j)%array_list(PMC_MAX_ARRAY))
[1762]	161	end do
	162
	163	CALL Get_DA_names_from_client (ClientId)
	164	end do
	165
	166	return
	167	END SUBROUTINE PMC_ServerInit
	168
	169	SUBROUTINE PMC_S_Set_2D_index_list (ClientId, index_list)
	170	IMPLICIT none
	171	INTEGER,INTENT(IN) :: ClientId
	172	INTEGER,DIMENSION(:,:),INTENT(INOUT) :: index_list !Index list will be modified in sort, therefore INOUT
	173
	174	!-- Local variables
	175	INTEGER :: ip,is,ie,ian,ic,n
	176	INTEGER :: istat
	177
	178	if(m_model_rank == 0) then
	179	CALL PMC_sort (index_list, 6) ! Sort to ascending Server PE
	180	is = 1
	181
	182	do ip=0,m_model_npes-1
	183
	184	! Split into Server PEs
	185	ie = is-1 !there may be no entry for this PE
	186	if(is <= size(index_list,2) .and. ie >= 0) then
	187	do while ( index_list(6,ie+1) == ip)
	188	ie = ie+1
	189	if( ie == size(index_list,2)) EXIT
	190	end do
	191
	192	ian = ie-is+1
	193	else
	194	is = -1
	195	ie = -2
	196	ian = 0
	197	end if
	198
	199	! Send data to other server PEs
	200
	201	if(ip == 0) then
	202	indClients(ClientId)%NrPoints = ian
	203	if(ian > 0) then
	204	ALLOCATE (indClients(ClientId)%index_list_2d(6,ian))
	205	indClients(ClientId)%index_list_2d(:,1:ian) = index_list(:,is:ie)
	206	end if
	207	else
	208	CALL MPI_Send (ian, 1, MPI_INTEGER, ip, 1000, m_model_comm, istat)
	209	if(ian > 0) then
	210	CALL MPI_Send (index_list(1,is), 6*ian, MPI_INTEGER, ip, 1001, &
	211	m_model_comm, istat)
	212	end if
	213	end if
	214	is = ie+1
	215	end do
	216	else
	217	CALL MPI_Recv (indClients(ClientId)%NrPoints, 1, MPI_INTEGER, 0, 1000, m_model_comm, &
	218	MPI_STATUS_IGNORE, istat)
	219	ian = indClients(ClientId)%NrPoints
	220	if(ian > 0) then
	221	ALLOCATE(indClients(ClientId)%index_list_2d(6,ian))
	222	CALL MPI_RECV (indClients(ClientId)%index_list_2d, 6*ian, MPI_INTEGER, 0, 1001, &
	223	m_model_comm, MPI_STATUS_IGNORE, istat)
	224	end if
	225	end if
	226
	227	CALL Set_PE_index_list (ClientId,Clients(ClientId),indClients(ClientId)%index_list_2d,indClients(ClientId)%NrPoints)
	228
	229	return
	230	END SUBROUTINE PMC_S_Set_2D_index_list
	231
[1779]	232	SUBROUTINE PMC_S_clear_next_array_list
	233	IMPLICIT none
[1762]	234
[1779]	235	next_array_in_list = 0
[1762]	236
[1779]	237	return
	238	END SUBROUTINE PMC_S_clear_next_array_list
[1762]	239
[1779]	240	! List handling is still required to get minimal interaction with pmc_interface
	241	logical function PMC_S_GetNextArray (ClientId, myName)
	242	INTEGER(iwp),INTENT(IN) :: ClientId
	243	CHARACTER(len=*),INTENT(OUT) :: myName
[1762]	244
[1779]	245	!-- local variables
	246	TYPE(PeDef),POINTER :: aPE
	247	TYPE(ArrayDef),POINTER :: ar
[1762]	248
[1779]	249	next_array_in_list = next_array_in_list+1
[1762]	250
[1779]	251	!-- Array Names are the same on all client PE, so take first PE to get the name
	252	aPE => Clients(ClientId)%PEs(1)
	253
	254	if(next_array_in_list > aPE%Nr_arrays) then
	255	PMC_S_GetNextArray = .false. ! all arrays done
	256	return
	257	end if
	258
	259	ar => aPE%array_list(next_array_in_list)
	260	myName = ar%name
	261
	262	PMC_S_GetNextArray = .true. ! Return true if legal array
	263	return
[1762]	264	END function PMC_S_GetNextArray
	265
[1766]	266	SUBROUTINE PMC_S_Set_DataArray_2d (ClientId, array, array_2 )
[1779]	267
[1762]	268	IMPLICIT none
[1779]	269
[1762]	270	INTEGER,INTENT(IN) :: ClientId
[1779]	271	REAL(wp), INTENT(IN), DIMENSION(:,:) :: array
	272	REAL(wp), INTENT(IN), DIMENSION(:,:), OPTIONAL :: array_2
	273
[1762]	274	INTEGER :: NrDims
	275	INTEGER,DIMENSION (4) :: dims
	276	TYPE(c_ptr) :: array_adr
[1766]	277	TYPE(c_ptr) :: second_adr
[1762]	278
	279	dims = 1
	280
	281	NrDims = 2
	282	dims(1) = size(array,1)
	283	dims(2) = size(array,2)
	284	array_adr = c_loc(array)
	285
[1766]	286	IF ( PRESENT( array_2 ) ) THEN
	287	second_adr = c_loc(array_2)
[1779]	288	CALL PMC_S_SetArray (ClientId, NrDims, dims, array_adr, second_adr = second_adr)
[1766]	289	ELSE
[1779]	290	CALL PMC_S_SetArray (ClientId, NrDims, dims, array_adr)
[1766]	291	ENDIF
[1762]	292
	293	return
	294	END SUBROUTINE PMC_S_Set_DataArray_2d
	295
[1766]	296	SUBROUTINE PMC_S_Set_DataArray_3d (ClientId, array, nz_cl, nz, array_2 )
[1779]	297
[1762]	298	IMPLICIT none
[1779]	299
[1762]	300	INTEGER,INTENT(IN) :: ClientId
[1779]	301	REAL(wp), INTENT(IN), DIMENSION(:,:,:) :: array
	302	REAL(wp), INTENT(IN), DIMENSION(:,:,:), OPTIONAL :: array_2
[1762]	303	INTEGER,INTENT(IN) :: nz_cl
	304	INTEGER,INTENT(IN) :: nz
[1779]	305
[1762]	306	INTEGER :: NrDims
	307	INTEGER,DIMENSION (4) :: dims
	308	TYPE(c_ptr) :: array_adr
[1766]	309	TYPE(c_ptr) :: second_adr
[1762]	310
	311	dims = 1
	312
	313	dims = 0
	314	NrDims = 3
	315	dims(1) = size(array,1)
	316	dims(2) = size(array,2)
	317	dims(3) = size(array,3)
	318	dims(4) = nz_cl+dims(1)-nz ! works for first dimension 1:nz and 0:nz+1
	319
	320	array_adr = c_loc(array)
	321
[1766]	322	!
	323	!-- In PALM's pointer version, two indices have to be stored internally.
	324	!-- The active address of the data array is set in swap_timelevel
	325	IF ( PRESENT( array_2 ) ) THEN
	326	second_adr = c_loc(array_2)
[1779]	327	CALL PMC_S_SetArray (ClientId, NrDims, dims, array_adr, second_adr = second_adr)
[1766]	328	ELSE
[1779]	329	CALL PMC_S_SetArray (ClientId, NrDims, dims, array_adr)
[1766]	330	ENDIF
[1762]	331
	332	return
	333	END SUBROUTINE PMC_S_Set_DataArray_3d
	334
	335	SUBROUTINE PMC_S_setInd_and_AllocMem (ClientId)
[1779]	336
	337	USE control_parameters, &
	338	ONLY: message_string
	339
[1762]	340	IMPLICIT none
[1779]	341
[1786]	342	!
	343	!-- Naming convention for appendices: _sc -> server to client transfer
	344	!-- _cs -> client to server transfer
	345	!-- Send -> Server to client transfer
	346	!-- Recv -> client to server transfer
[1762]	347	INTEGER,INTENT(IN) :: ClientId
	348
[1779]	349	INTEGER :: i, istat, ierr, j
[1762]	350	INTEGER :: arlen, myIndex, tag
	351	INTEGER :: rCount ! count MPI requests
[1764]	352	INTEGER(idp) :: bufsize ! Size of MPI data Window
[1762]	353	TYPE(PeDef),POINTER :: aPE
	354	TYPE(ArrayDef),POINTER :: ar
	355	CHARACTER(len=DA_Namelen) :: myName
	356	INTEGER,DIMENSION(1024) :: req
	357	Type(c_ptr) :: base_ptr
[1786]	358	REAL(wp),DIMENSION(:),POINTER,SAVE :: base_array_sc !Base array for server to client transfer
	359	REAL(wp),DIMENSION(:),POINTER,SAVE :: base_array_cs !Base array for client to server transfer
[1762]	360	INTEGER(KIND=MPI_ADDRESS_KIND) :: WinSize
	361
[1786]	362	!
	363	!-- Server to client direction
[1762]	364	myIndex = 1
	365	rCount = 0
	366	bufsize = 8
	367
[1786]	368	!
	369	!-- First stride: Compute size and set index
[1762]	370	do i=1,Clients(ClientId)%inter_npes
	371	aPE => Clients(ClientId)%PEs(i)
	372	tag = 200
[1779]	373	do j=1,aPE%Nr_arrays
	374	ar => aPE%array_list(j)
	375	if(ar%NrDims == 2) then
[1786]	376	arlen = aPE%NrEle ! 2D
[1779]	377	else if(ar%NrDims == 3) then
[1786]	378	arlen = aPE%NrEle * ar%A_dim(4); ! 3D
[1762]	379	else
	380	arlen = -1
	381	end if
[1786]	382	ar%SendIndex = myIndex
[1762]	383
	384	tag = tag+1
	385	rCount = rCount+1
	386	CALL MPI_Isend (myIndex, 1, MPI_INTEGER, i-1, tag, Clients(ClientId)%inter_comm, req(rCount),ierr)
	387
	388	if(rCount == 1024) then ! Maximum of 1024 outstanding requests
	389	CALL MPI_Waitall (rCount, req, MPI_STATUSES_IGNORE, ierr)
	390	rCount = 0;
	391	end if
	392
	393	myIndex = myIndex+arlen
	394	bufsize = bufsize+arlen
[1786]	395	ar%SendSize = arlen
[1762]	396
	397	end do
[1786]	398	if(rCount > 0) then ! Wait for all send completed
[1762]	399	CALL MPI_Waitall (rCount, req, MPI_STATUSES_IGNORE, ierr)
	400	end if
	401	end do
	402
[1786]	403	!
	404	!-- Create RMA (One Sided Communication) window for data buffer server to
	405	!-- client transfer.
	406	!-- The buffer of MPI_Get (counter part of transfer) can be PE-local, i.e.
	407	!-- it can but must not be part of the MPI RMA window.
	408	!-- Only one RMA window is required to prepare the data
	409	!-- for server -> client transfer on the server side and
	410	!-- for client -> server transfer on the client side
	411	CALL PMC_Alloc_mem (base_array_sc, bufsize)
	412	Clients(ClientId)%TotalBufferSize = bufsize*wp !Total buffer size in Byte
[1762]	413
[1786]	414	WinSize = bufsize*wp
	415	CALL MPI_Win_create (base_array_sc, WinSize, wp, MPI_INFO_NULL, &
	416	Clients(ClientId)%intra_comm, Clients(ClientId)%win_server_client, ierr)
	417	CALL MPI_Win_fence (0, Clients(ClientId)%win_server_client, ierr); ! Open Window to set data
	418	!
	419	!-- Second stride: Set Buffer pointer
	420	do i=1,Clients(ClientId)%inter_npes
	421	aPE => Clients(ClientId)%PEs(i)
	422	do j=1,aPE%Nr_arrays
	423	ar => aPE%array_list(j)
	424	ar%SendBuf = c_loc(base_array_sc(ar%SendIndex))
	425	if(ar%SendIndex+ar%SendSize > bufsize) then
	426	write(0,'(a,i4,4i7,1x,a)') 'Server Buffer too small ',i,ar%SendIndex,ar%SendSize,ar%SendIndex+ar%SendSize,bufsize,trim(ar%name)
	427	CALL MPI_Abort (MPI_COMM_WORLD, istat, ierr)
	428	end if
	429	end do
	430	end do
	431
	432	!-- Client to server direction
	433
	434	bufsize = 8
	435
	436	!-- First stride, Compute size and set index
	437
	438	do i=1,Clients(ClientId)%inter_npes
	439	aPE => Clients(ClientId)%PEs(i)
	440	tag = 300
	441
	442	do j=1,aPE%Nr_arrays
	443	ar => aPE%array_list(j)
	444
	445	! Receive Index from client
	446	tag = tag+1
	447	CALL MPI_Recv (myIndex, 1, MPI_INTEGER, i-1, tag, Clients(ClientId)%inter_comm, MPI_STATUS_IGNORE, ierr)
	448
	449	if(ar%NrDims == 3) then
	450	bufsize = max(bufsize,aPE%NrEle * ar%A_dim(4)) ! 3D
	451	else
	452	bufsize = max(bufsize,aPE%NrEle) ! 2D
	453	end if
	454	ar%RecvIndex = myIndex
	455	end do
	456
	457	end do
	458
	459	!-- Create RMA (One Sided Communication) data buffer
	460	!-- The buffer for MPI_Get can be PE local, i.e. it can but must not be part of the MPI RMA window
	461
	462	CALL PMC_Alloc_mem (base_array_cs, bufsize, base_ptr)
[1762]	463	Clients(ClientId)%TotalBufferSize = bufsize*wp !Total buffer size in Byte
	464
	465	CALL MPI_Barrier(Clients(ClientId)%intra_comm, ierr)
	466
[1786]	467	!-- Second stride, Set Buffer pointer
[1762]	468
	469	do i=1,Clients(ClientId)%inter_npes
	470	aPE => Clients(ClientId)%PEs(i)
[1786]	471
[1779]	472	do j=1,aPE%Nr_arrays
	473	ar => aPE%array_list(j)
[1786]	474	ar%RecvBuf = base_ptr
[1762]	475	end do
	476	end do
	477
	478	return
	479	END SUBROUTINE PMC_S_setInd_and_AllocMem
	480
	481	SUBROUTINE PMC_S_FillBuffer (ClientId, WaitTime)
	482	IMPLICIT none
[1779]	483	INTEGER,INTENT(IN) :: ClientId
	484	REAL(wp), INTENT(OUT), OPTIONAL :: WaitTime
[1762]	485
[1779]	486	INTEGER :: ip,ij,istat,ierr,j
	487	INTEGER :: myIndex
	488	REAL(wp) :: t1,t2
	489	TYPE(PeDef),POINTER :: aPE
	490	TYPE(ArrayDef),POINTER :: ar
	491	CHARACTER(len=DA_Namelen) :: myName
	492	INTEGER,DIMENSION(1) :: buf_shape
	493	REAL(wp), POINTER, DIMENSION(:) :: buf
	494	REAL(wp), POINTER, DIMENSION(:,:) :: data_2d
	495	REAL(wp), POINTER, DIMENSION(:,:,:) :: data_3d
[1762]	496
	497	t1 = PMC_Time()
	498	CALL MPI_Barrier(Clients(ClientId)%intra_comm, ierr) ! Wait for buffer empty
	499	t2 = PMC_Time()
	500	if(present(WaitTime)) WaitTime = t2-t1
	501
	502	do ip=1,Clients(ClientId)%inter_npes
	503	aPE => Clients(ClientId)%PEs(ip)
[1779]	504	do j=1,aPE%Nr_arrays
	505	ar => aPE%array_list(j)
[1762]	506	myIndex=1
[1779]	507	if(ar%NrDims == 2) then
[1762]	508	buf_shape(1) = aPE%NrEle
	509	CALL c_f_pointer(ar%SendBuf, buf, buf_shape)
	510	CALL c_f_pointer(ar%data, data_2d, ar%A_dim(1:2))
	511	do ij=1,aPE%NrEle
	512	buf(myIndex) = data_2d(aPE%locInd(ij)%j,aPE%locInd(ij)%i)
	513	myIndex = myIndex+1
	514	end do
[1779]	515	else if(ar%NrDims == 3) then
[1762]	516	buf_shape(1) = aPE%NrEle*ar%A_dim(4)
	517	CALL c_f_pointer(ar%SendBuf, buf, buf_shape)
	518	CALL c_f_pointer(ar%data, data_3d, ar%A_dim(1:3))
	519	do ij=1,aPE%NrEle
	520	buf(myIndex:myIndex+ar%A_dim(4)-1) = data_3d(1:ar%A_dim(4),aPE%locInd(ij)%j,aPE%locInd(ij)%i)
	521	myIndex = myIndex+ar%A_dim(4)
	522	end do
	523	end if
	524	end do
	525	end do
	526
[1786]	527	CALL MPI_Barrier(Clients(ClientId)%intra_comm, ierr) ! buffer is filled
[1762]	528
	529	return
	530	END SUBROUTINE PMC_S_FillBuffer
	531
	532	SUBROUTINE PMC_S_GetData_from_Buffer (ClientId, WaitTime)
[1779]	533
[1762]	534	IMPLICIT none
	535
[1779]	536	INTEGER,INTENT(IN) :: ClientId
	537	REAL(wp), INTENT(OUT), OPTIONAL :: WaitTime
	538
[1762]	539	!-- local variables
[1779]	540	INTEGER :: ip,ij,istat,ierr,j
	541	INTEGER :: myIndex
[1786]	542	INTEGER :: nr
[1779]	543	REAL(wp) :: t1,t2
	544	TYPE(PeDef),POINTER :: aPE
	545	TYPE(ArrayDef),POINTER :: ar
	546	CHARACTER(len=DA_Namelen) :: myName
	547	INTEGER,DIMENSION(1) :: buf_shape
	548	REAL(wp), POINTER, DIMENSION(:) :: buf
	549	REAL(wp), POINTER, DIMENSION(:,:) :: data_2d
	550	REAL(wp), POINTER, DIMENSION(:,:,:) :: data_3d
[1786]	551	INTEGER :: target_pe
	552	INTEGER(kind=MPI_ADDRESS_KIND) :: target_disp
[1762]	553
	554	t1 = PMC_Time()
[1786]	555	CALL MPI_Barrier(Clients(ClientId)%intra_comm, ierr) ! Wait for client to fill buffer
	556	t2 = PMC_Time()-t1
	557	if(present(WaitTime)) WaitTime = t2
[1762]	558
[1786]	559	! CALL MPI_Win_fence (0, Clients(ClientId)%win_server_client, ierr) ! Fence might do it, test later
	560	CALL MPI_Barrier(Clients(ClientId)%intra_comm, ierr) ! Wait for buffer is filled
	561
[1762]	562	do ip=1,Clients(ClientId)%inter_npes
	563	aPE => Clients(ClientId)%PEs(ip)
[1779]	564	do j=1,aPE%Nr_arrays
	565	ar => aPE%array_list(j)
[1786]	566
	567	if(ar%RecvIndex < 0) CYCLE
	568
	569	if(ar%NrDims == 2) then
	570	nr = aPE%NrEle
	571	else if(ar%NrDims == 3) then
	572	nr = aPE%NrEle*ar%A_dim(4)
	573	end if
	574
	575	buf_shape(1) = nr
	576	CALL c_f_pointer(ar%RecvBuf, buf, buf_shape)
	577	!
	578	!-- MPI passive target RMA
	579
	580	if(nr > 0) then
	581	target_disp = ar%RecvIndex-1
	582	target_pe = ip-1+m_model_npes ! client PEs are located behind server PEs
	583	CALL MPI_Win_lock (MPI_LOCK_SHARED , target_pe, 0, Clients(ClientId)%win_server_client, ierr)
	584	CALL MPI_Get (buf, nr, MPI_REAL, target_pe, target_disp, nr, MPI_REAL, Clients(ClientId)%win_server_client, ierr)
	585	CALL MPI_Win_unlock (target_pe, Clients(ClientId)%win_server_client, ierr)
	586	end if
	587
[1762]	588	myIndex=1
[1779]	589	if(ar%NrDims == 2) then
[1762]	590	CALL c_f_pointer(ar%data, data_2d, ar%A_dim(1:2))
	591	do ij=1,aPE%NrEle
	592	data_2d(aPE%locInd(ij)%j,aPE%locInd(ij)%i) = buf(myIndex)
	593	myIndex = myIndex+1
	594	end do
[1779]	595	else if(ar%NrDims == 3) then
[1762]	596	CALL c_f_pointer(ar%data, data_3d, ar%A_dim(1:3))
	597	do ij=1,aPE%NrEle
	598	data_3d(1:ar%A_dim(4),aPE%locInd(ij)%j,aPE%locInd(ij)%i) = buf(myIndex:myIndex+ar%A_dim(4)-1)
	599	myIndex = myIndex+ar%A_dim(4)
	600	end do
	601	end if
	602	end do
	603	end do
	604
	605	END SUBROUTINE PMC_S_GetData_from_Buffer
	606
	607	! Private SUBROUTINEs
	608
	609	SUBROUTINE Get_DA_names_from_client (ClientId)
	610	IMPLICIT none
	611	INTEGER,INTENT(IN) :: ClientId
	612	!-- local variables
	613	type(DA_NameDef) :: myName
	614
	615	! Get Data Array Description and Name from Client
	616
	617	do
	618	CALL PMC_Bcast ( myName%couple_index, 0, comm=m_to_client_comm(ClientId))
	619	if(myName%couple_index == -1) EXIT
	620	CALL PMC_Bcast ( myName%ServerDesc, 0, comm=m_to_client_comm(ClientId))
	621	CALL PMC_Bcast ( myName%NameOnServer, 0, comm=m_to_client_comm(ClientId))
	622	CALL PMC_Bcast ( myName%ClientDesc, 0, comm=m_to_client_comm(ClientId))
	623	CALL PMC_Bcast ( myName%NameOnClient, 0, comm=m_to_client_comm(ClientId))
	624
	625	CALL PMC_G_SetName (clients(ClientID), myName%couple_index, myName%NameOnServer )
	626	end do
	627
	628	return
	629	END SUBROUTINE Get_DA_names_from_client
	630
[1779]	631	SUBROUTINE PMC_S_SetArray (ClientId, NrDims, dims, array_adr, second_adr)
[1762]	632	IMPLICIT none
	633
	634	INTEGER,INTENT(IN) :: ClientId
	635	INTEGER,INTENT(IN) :: NrDims
	636	INTEGER,INTENT(IN),DIMENSION(:) :: dims
	637	TYPE(c_ptr),INTENT(IN) :: array_adr
[1766]	638	TYPE(c_ptr),INTENT(IN),OPTIONAL :: second_adr
[1762]	639
	640	INTEGER :: i
	641	TYPE(PeDef),POINTER :: aPE
	642	TYPE(ArrayDef),POINTER :: ar
	643	CHARACTER(len=DA_Namelen) :: myName
	644
	645	! Set Array for Client interPE 0
	646
	647	do i=1,Clients(ClientId)%inter_npes
	648	aPE => Clients(ClientId)%PEs(i)
[1779]	649	ar => aPE%array_list(next_array_in_list)
[1762]	650	ar%NrDims = NrDims
	651	ar%A_dim = dims
	652	ar%data = array_adr
[1766]	653	if(present(second_adr)) then
	654	ar%po_data(1) = array_adr
	655	ar%po_data(2) = second_adr
	656	else
	657	ar%po_data(1) = C_NULL_PTR
	658	ar%po_data(2) = C_NULL_PTR
	659	end if
[1762]	660	end do
	661
	662	return
	663	END SUBROUTINE PMC_S_SetArray
	664
	665
[1766]	666	SUBROUTINE PMC_S_Set_Active_data_array (ClientId,iactive)
	667	IMPLICIT none
	668
	669	INTEGER,INTENT(IN) :: ClientId
	670	INTEGER,INTENT(IN) :: iactive
	671
	672	!-- local variables
[1779]	673	INTEGER :: i, ip, j
[1766]	674	TYPE(PeDef),POINTER :: aPE
	675	TYPE(ArrayDef),POINTER :: ar
	676	CHARACTER(len=DA_Namelen) :: myName
	677
	678	do ip=1,Clients(ClientId)%inter_npes
	679	aPE => Clients(ClientId)%PEs(ip)
[1779]	680	do j=1,aPE%Nr_arrays
	681	ar => aPE%array_list(j)
[1766]	682	if(iactive == 1 .OR. iactive == 2) then
	683	ar%data = ar%po_data(iactive)
	684	end if
	685	end do
	686	end do
	687
	688	return
	689	END SUBROUTINE PMC_S_Set_Active_data_array
	690
	691
[1762]	692	SUBROUTINE Set_PE_index_list (ClientId, myClient,index_list,NrP)
	693	IMPLICIT none
	694
	695	INTEGER,INTENT(IN) :: ClientId
	696	TYPE(ClientDef),INTENT(INOUT) :: myClient
	697	INTEGER,INTENT(IN),DIMENSION(:,:) :: index_list
	698	INTEGER,INTENT(IN) :: NrP
	699
	700	!-- local variables
	701	INTEGER :: i,j,ind,ierr,i2
	702	TYPE(PeDef),POINTER :: aPE
	703	INTEGER :: RemPE
	704	INTEGER,DIMENSION(myClient%inter_npes) :: RemInd
	705	INTEGER,DIMENSION(:),POINTER :: RemIndw
	706	INTEGER,DIMENSION(:),POINTER :: RLdef
	707	INTEGER(KIND=MPI_ADDRESS_KIND) :: WinSize
	708	INTEGER :: indWin,indWin2
	709
	710	! First, count entries for every remote client PE
	711
	712	do i=1,myClient%inter_npes
	713	aPE => myClient%PEs(i)
	714	aPE%NrEle = 0
	715	end do
	716
	717	do j=1,NrP ! loop over number of cells coarse grid
	718	RemPE = index_list(5,j)+1 ! Pe number remote PE
	719	aPE => myClient%PEs(RemPE)
	720	aPE% NrEle = aPE% NrEle+1 ! Increment Number of elements for this client Pe
	721	end do
	722
	723	do i=1,myClient%inter_npes
	724	aPE => myClient%PEs(i)
	725	ALLOCATE(aPE%locInd(aPE%NrEle))
	726	end do
	727
	728	RemInd = 0
	729
	730	! Second, Create lists
	731
	732	do j=1,NrP ! loop over number of cells coarse grid
	733	RemPE = index_list(5,j)+1 ! Pe number remote PE
	734	aPE => myClient%PEs(RemPE)
	735	RemInd(RemPE) = RemInd(RemPE)+1
	736	ind = RemInd(RemPE)
	737	aPE%locInd(ind)%i = index_list(1,j)
	738	aPE%locInd(ind)%j = index_list(2,j)
	739	end do
	740
	741	! Prepare Number of Elements for Client PEs
	742	CALL PMC_Alloc_mem (RLdef, myClient%inter_npes*2)
	743	WinSize = myClient%inter_npesc_sizeof(i)2 ! Number of Client PEs * size of INTEGER (i just arbitrary INTEGER)
	744
	745	CALL MPI_Win_create (RLdef, WinSize, iwp, MPI_INFO_NULL, myClient%intra_comm, indWin, ierr);
	746	CALL MPI_Win_fence (0, indWin, ierr); ! Open Window to set data
	747
	748	RLdef(1) = 0 ! Index on Remote PE 0
	749	RLdef(2) = RemInd(1) ! Number of Elements on Rem PE 0
	750
	751	do i=2,myClient%inter_npes ! Reserve Buffer for index array
	752	i2 = (i-1)*2+1
	753	RLdef(i2) = RLdef(i2-2) + RLdef(i2-1)*2 ! Index on Remote PE
	754	RLdef(i2+1) = RemInd(i) ! Number of Elements on Remote PE
	755	end do
	756
	757	CALL MPI_Win_fence (0, indWin, ierr); ! Close Window to allow client to access data
	758	CALL MPI_Win_fence (0, indWin, ierr); ! Client has retrieved data
	759
	760	i2 = 2*myClient%inter_npes-1
	761	WinSize = (RLdef(i2)+RLdef(i2+1))*2
	762	WinSize = max(WinSize,1) ! Make sure, MPI_Alloc_mem works
	763
	764	CALL PMC_Alloc_mem (RemIndw, int(WinSize))
	765
	766	CALL MPI_Barrier (m_model_comm, ierr)
	767	CALL MPI_Win_create (RemIndw, WinSize*c_sizeof(i), iwp, MPI_INFO_NULL, myClient%intra_comm, indWin2, ierr);
	768
	769	CALL MPI_Win_fence (0, indWin2, ierr); ! Open Window to set data
	770	do j=1,NrP ! this loop creates the 2D index list
	771	RemPE = index_list(5,j)+1 ! Pe number remote PE
	772	aPE => myClient%PEs(RemPE)
	773	i2 = RemPE*2-1
	774	ind = RLdef(i2)+1
	775	RemIndw(ind) = index_list(3,j)
	776	RemIndw(ind+1) = index_list(4,j)
	777	RLdef(i2) = RLdef(i2)+2
	778	end do
	779	CALL MPI_Win_fence (0, indWin2, ierr); !all data set
	780
	781	CALL MPI_Barrier(myClient%intra_comm, ierr) ! Dont know why, but this barrier is necessary before we can free the windows
	782
	783	CALL MPI_Win_free(indWin, ierr)
	784	CALL MPI_Win_free(indWin2, ierr)
	785
	786	! Sollte funktionieren, Problem mit MPI implementation
	787	! https://www.lrz.de/services/software/parallel/mpi/onesided
	788	! CALL MPI_Free_mem (RemIndw, ierr)
	789
	790	return
	791	END SUBROUTINE Set_PE_index_list
	792
[1764]	793	#endif
[1762]	794	END MODULE pmc_server

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

Download in other formats:

| Impressum | ©Leibniz Universität Hannover |