Context Navigation

mod_hallo.F90 @ 1

Last change on this file since 1 was 1, checked in by emillour, 14 years ago
Import initial LMDZ5
File size: 21.3 KB

Rev	Line
[1]	1	module mod_Hallo
	2	USE parallel
	3	implicit none
	4	logical,save :: use_mpi_alloc
	5	integer, parameter :: MaxRequest=200
	6	integer, parameter :: MaxProc=80
	7	integer, parameter :: MaxBufferSize=1024102416
	8	integer, parameter :: ListSize=1000
	9
	10	integer,save :: MaxBufferSize_Used
	11	!$OMP THREADPRIVATE( MaxBufferSize_Used)
	12
	13	real,save,pointer,dimension(:) :: Buffer
	14	!$OMP THREADPRIVATE(Buffer)
	15
	16	integer,save,dimension(Listsize) :: Buffer_Pos
	17	integer,save :: Index_Pos
	18	!$OMP THREADPRIVATE(Buffer_Pos,Index_pos)
	19
	20	type Hallo
	21	real, dimension(:,:),pointer :: Field
	22	integer :: offset
	23	integer :: size
	24	integer :: NbLevel
	25	integer :: Stride
	26	end type Hallo
	27
	28	type request_SR
	29	integer :: NbRequest=0
	30	integer :: Pos
	31	integer :: Index
	32	type(Hallo),dimension(MaxRequest) :: Hallo
	33	integer :: MSG_Request
	34	end type request_SR
	35
	36	type request
	37	type(request_SR),dimension(0:MaxProc-1) :: RequestSend
	38	type(request_SR),dimension(0:MaxProc-1) :: RequestRecv
	39	integer :: tag=1
	40	end type request
	41
	42
	43	contains
	44
	45	subroutine Init_mod_hallo
	46	implicit none
	47
	48	Index_Pos=1
	49	Buffer_Pos(Index_Pos)=1
	50	MaxBufferSize_Used=0
	51
	52	IF (use_mpi_alloc .AND. using_mpi) THEN
	53	CALL create_global_mpi_buffer
	54	ELSE
	55	CALL create_standard_mpi_buffer
	56	ENDIF
	57
	58	end subroutine init_mod_hallo
	59
	60	SUBROUTINE create_standard_mpi_buffer
	61	IMPLICIT NONE
	62
	63	ALLOCATE(Buffer(MaxBufferSize))
	64
	65	END SUBROUTINE create_standard_mpi_buffer
	66
	67	SUBROUTINE create_global_mpi_buffer
	68	IMPLICIT NONE
	69	#ifdef CPP_MPI
	70	INCLUDE 'mpif.h'
	71	#endif
	72	POINTER (Pbuffer,MPI_Buffer(MaxBufferSize))
	73	REAL :: MPI_Buffer
	74	#ifdef CPP_MPI
	75	INTEGER(KIND=MPI_ADDRESS_KIND) :: BS
	76	#else
	77	INTEGER(KIND=8) :: BS
	78	#endif
	79	INTEGER :: i,ierr
	80
	81	! Allocation du buffer MPI
	82	Bs=8*MaxBufferSize
	83	!$OMP CRITICAL (MPI)
	84	#ifdef CPP_MPI
	85	CALL MPI_ALLOC_MEM(BS,MPI_INFO_NULL,Pbuffer,ierr)
	86	#endif
	87	!$OMP END CRITICAL (MPI)
	88	DO i=1,MaxBufferSize
	89	MPI_Buffer(i)=i
	90	ENDDO
	91
	92	CALL Associate_buffer(MPI_Buffer)
	93
	94	CONTAINS
	95
	96	SUBROUTINE Associate_buffer(MPI_Buffer)
	97	IMPLICIT NONE
	98	REAL,DIMENSION(:),target :: MPI_Buffer
	99
	100	Buffer=>MPI_Buffer
	101
	102	END SUBROUTINE Associate_buffer
	103
	104	END SUBROUTINE create_global_mpi_buffer
	105
	106
	107	subroutine allocate_buffer(Size,Index,Pos)
	108	implicit none
	109	integer :: Size
	110	integer :: Index
	111	integer :: Pos
	112
	113	if (Buffer_pos(Index_pos)+Size>MaxBufferSize_Used) MaxBufferSize_Used=Buffer_pos(Index_pos)+Size
	114	if (Buffer_pos(Index_pos)+Size>MaxBufferSize) then
	115	print *,'STOP :: La taille de MaxBufferSize dans mod_hallo.F90 est trop petite !!!!'
	116	stop
	117	endif
	118
	119	if (Index_pos>=ListSize) then
	120	print *,'STOP :: La taille de ListSize dans mod_hallo.F90 est trop petite !!!!'
	121	stop
	122	endif
	123
	124	Pos=Buffer_Pos(Index_Pos)
	125	Buffer_Pos(Index_pos+1)=Buffer_Pos(Index_Pos)+Size
	126	Index_Pos=Index_Pos+1
	127	Index=Index_Pos
	128
	129	end subroutine allocate_buffer
	130
	131	subroutine deallocate_buffer(Index)
	132	implicit none
	133	integer :: Index
	134
	135	Buffer_Pos(Index)=-1
	136
	137	do while (Buffer_Pos(Index_Pos)==-1 .and. Index_Pos>1)
	138	Index_Pos=Index_Pos-1
	139	end do
	140
	141	end subroutine deallocate_buffer
	142
	143	subroutine SetTag(a_request,tag)
	144	implicit none
	145	type(request):: a_request
	146	integer :: tag
	147
	148	a_request%tag=tag
	149	end subroutine SetTag
	150
	151
	152	subroutine Init_Hallo(Field,Stride,NbLevel,offset,size,NewHallo)
	153	integer :: Stride
	154	integer :: NbLevel
	155	integer :: size
	156	integer :: offset
	157	real, dimension(Stride,NbLevel),target :: Field
	158	type(Hallo) :: NewHallo
	159
	160	NewHallo%Field=>Field
	161	NewHallo%Stride=Stride
	162	NewHallo%NbLevel=NbLevel
	163	NewHallo%size=size
	164	NewHallo%offset=offset
	165
	166
	167	end subroutine Init_Hallo
	168
	169	subroutine Register_SendField(Field,ij,ll,offset,size,target,a_request)
	170	implicit none
	171
	172	#include "dimensions.h"
	173	#include "paramet.h"
	174
	175	INTEGER :: ij,ll,offset,size,target
	176	REAL, dimension(ij,ll) :: Field
	177	type(request),target :: a_request
	178	type(request_SR),pointer :: Ptr_request
	179
	180	Ptr_Request=>a_request%RequestSend(target)
	181	Ptr_Request%NbRequest=Ptr_Request%NbRequest+1
	182	if (Ptr_Request%NbRequest>=MaxRequest) then
	183	print *,'STOP :: La taille de MaxRequest dans mod_hallo.F90 est trop petite !!!!'
	184	stop
	185	endif
	186	call Init_Hallo(Field,ij,ll,offset,size,Ptr_request%Hallo(Ptr_Request%NbRequest))
	187
	188	end subroutine Register_SendField
	189
	190	subroutine Register_RecvField(Field,ij,ll,offset,size,target,a_request)
	191	implicit none
	192
	193	#include "dimensions.h"
	194	#include "paramet.h"
	195
	196	INTEGER :: ij,ll,offset,size,target
	197	REAL, dimension(ij,ll) :: Field
	198	type(request),target :: a_request
	199	type(request_SR),pointer :: Ptr_request
	200
	201	Ptr_Request=>a_request%RequestRecv(target)
	202	Ptr_Request%NbRequest=Ptr_Request%NbRequest+1
	203
	204	if (Ptr_Request%NbRequest>=MaxRequest) then
	205	print *,'STOP :: La taille de MaxRequest dans mod_hallo.F90 est trop petite !!!!'
	206	stop
	207	endif
	208
	209	call Init_Hallo(Field,ij,ll,offset,size,Ptr_request%Hallo(Ptr_Request%NbRequest))
	210
	211
	212	end subroutine Register_RecvField
	213
	214	subroutine Register_SwapField(FieldS,FieldR,ij,ll,jj_Nb_New,a_request)
	215
	216	implicit none
	217	#include "dimensions.h"
	218	#include "paramet.h"
	219
	220	INTEGER :: ij,ll
	221	REAL, dimension(ij,ll) :: FieldS
	222	REAL, dimension(ij,ll) :: FieldR
	223	type(request) :: a_request
	224	integer,dimension(0:MPI_Size-1) :: jj_Nb_New
	225	integer,dimension(0:MPI_Size-1) :: jj_Begin_New,jj_End_New
	226
	227	integer ::i,jje,jjb
	228
	229	jj_begin_New(0)=1
	230	jj_End_New(0)=jj_Nb_New(0)
	231	do i=1,MPI_Size-1
	232	jj_begin_New(i)=jj_end_New(i-1)+1
	233	jj_end_New(i)=jj_begin_new(i)+jj_Nb_New(i)-1
	234	enddo
	235
	236	do i=0,MPI_Size-1
	237	if (i /= MPI_Rank) then
	238	jjb=max(jj_begin_new(i),jj_begin)
	239	jje=min(jj_end_new(i),jj_end)
	240
	241	if (ij==ip1jm .and. jje==jjp1) jje=jjm
	242
	243	if (jje >= jjb) then
	244	call Register_SendField(FieldS,ij,ll,jjb,jje-jjb+1,i,a_request)
	245	endif
	246
	247	jjb=max(jj_begin_new(MPI_Rank),jj_begin_Para(i))
	248	jje=min(jj_end_new(MPI_Rank),jj_end_Para(i))
	249
	250	if (ij==ip1jm .and. jje==jjp1) jje=jjm
	251
	252	if (jje >= jjb) then
	253	call Register_RecvField(FieldR,ij,ll,jjb,jje-jjb+1,i,a_request)
	254	endif
	255
	256	endif
	257	enddo
	258
	259	end subroutine Register_SwapField
	260
	261
	262	subroutine Register_SwapFieldHallo(FieldS,FieldR,ij,ll,jj_Nb_New,Up,Down,a_request)
	263
	264	implicit none
	265	#include "dimensions.h"
	266	#include "paramet.h"
	267
	268	INTEGER :: ij,ll,Up,Down
	269	REAL, dimension(ij,ll) :: FieldS
	270	REAL, dimension(ij,ll) :: FieldR
	271	type(request) :: a_request
	272	integer,dimension(0:MPI_Size-1) :: jj_Nb_New
	273	integer,dimension(0:MPI_Size-1) :: jj_Begin_New,jj_End_New
	274
	275	integer ::i,jje,jjb
	276
	277	jj_begin_New(0)=1
	278	jj_End_New(0)=jj_Nb_New(0)
	279	do i=1,MPI_Size-1
	280	jj_begin_New(i)=jj_end_New(i-1)+1
	281	jj_end_New(i)=jj_begin_new(i)+jj_Nb_New(i)-1
	282	enddo
	283
	284	do i=0,MPI_Size-1
	285	jj_begin_New(i)=max(1,jj_begin_New(i)-Up)
	286	jj_end_New(i)=min(jjp1,jj_end_new(i)+Down)
	287	enddo
	288
	289	do i=0,MPI_Size-1
	290	if (i /= MPI_Rank) then
	291	jjb=max(jj_begin_new(i),jj_begin)
	292	jje=min(jj_end_new(i),jj_end)
	293
	294	if (ij==ip1jm .and. jje==jjp1) jje=jjm
	295
	296	if (jje >= jjb) then
	297	call Register_SendField(FieldS,ij,ll,jjb,jje-jjb+1,i,a_request)
	298	endif
	299
	300	jjb=max(jj_begin_new(MPI_Rank),jj_begin_Para(i))
	301	jje=min(jj_end_new(MPI_Rank),jj_end_Para(i))
	302
	303	if (ij==ip1jm .and. jje==jjp1) jje=jjm
	304
	305	if (jje >= jjb) then
	306	call Register_RecvField(FieldR,ij,ll,jjb,jje-jjb+1,i,a_request)
	307	endif
	308
	309	endif
	310	enddo
	311
	312	end subroutine Register_SwapFieldHallo
	313
	314	subroutine Register_Hallo(Field,ij,ll,RUp,Rdown,SUp,SDown,a_request)
	315
	316	implicit none
	317	#include "dimensions.h"
	318	#include "paramet.h"
	319	#ifdef CPP_MPI
	320	include 'mpif.h'
	321	#endif
	322	INTEGER :: ij,ll
	323	REAL, dimension(ij,ll) :: Field
	324	INTEGER :: Sup,Sdown,rup,rdown
	325	type(request) :: a_request
	326	type(Hallo),pointer :: PtrHallo
	327	LOGICAL :: SendUp,SendDown
	328	LOGICAL :: RecvUp,RecvDown
	329
	330
	331	SendUp=.TRUE.
	332	SendDown=.TRUE.
	333	RecvUp=.TRUE.
	334	RecvDown=.TRUE.
	335
	336	IF (pole_nord) THEN
	337	SendUp=.FALSE.
	338	RecvUp=.FALSE.
	339	ENDIF
	340
	341	IF (pole_sud) THEN
	342	SendDown=.FALSE.
	343	RecvDown=.FALSE.
	344	ENDIF
	345
	346	if (Sup.eq.0) then
	347	SendUp=.FALSE.
	348	endif
	349
	350	if (Sdown.eq.0) then
	351	SendDown=.FALSE.
	352	endif
	353
	354	if (Rup.eq.0) then
	355	RecvUp=.FALSE.
	356	endif
	357
	358	if (Rdown.eq.0) then
	359	RecvDown=.FALSE.
	360	endif
	361
	362	IF (SendUp) THEN
	363	call Register_SendField(Field,ij,ll,jj_begin,SUp,MPI_Rank-1,a_request)
	364	ENDIF
	365
	366	IF (SendDown) THEN
	367	call Register_SendField(Field,ij,ll,jj_end-SDown+1,SDown,MPI_Rank+1,a_request)
	368	ENDIF
	369
	370
	371	IF (RecvUp) THEN
	372	call Register_RecvField(Field,ij,ll,jj_begin-Rup,RUp,MPI_Rank-1,a_request)
	373	ENDIF
	374
	375	IF (RecvDown) THEN
	376	call Register_RecvField(Field,ij,ll,jj_end+1,RDown,MPI_Rank+1,a_request)
	377	ENDIF
	378
	379	end subroutine Register_Hallo
	380
	381	subroutine SendRequest(a_Request)
	382	implicit none
	383
	384	#include "dimensions.h"
	385	#include "paramet.h"
	386	#ifdef CPP_MPI
	387	include 'mpif.h'
	388	#endif
	389
	390	type(request),target :: a_request
	391	type(request_SR),pointer :: Req
	392	type(Hallo),pointer :: PtrHallo
	393	integer :: SizeBuffer
	394	integer :: i,rank,l,ij,Pos,ierr
	395	integer :: offset
	396	real,dimension(:,:),pointer :: Field
	397	integer :: Nb
	398
	399	do rank=0,MPI_SIZE-1
	400
	401	Req=>a_Request%RequestSend(rank)
	402
	403	SizeBuffer=0
	404	do i=1,Req%NbRequest
	405	PtrHallo=>Req%Hallo(i)
	406	!$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
	407	DO l=1,PtrHallo%NbLevel
	408	SizeBuffer=SizeBuffer+PtrHallo%size*iip1
	409	ENDDO
	410	!$OMP ENDDO NOWAIT
	411	enddo
	412
	413	if (SizeBuffer>0) then
	414
	415	call allocate_buffer(SizeBuffer,Req%Index,Req%pos)
	416
	417	Pos=Req%Pos
	418	do i=1,Req%NbRequest
	419	PtrHallo=>Req%Hallo(i)
	420	offset=(PtrHallo%offset-1)*iip1+1
	421	Nb=iip1*PtrHallo%size-1
	422	Field=>PtrHallo%Field
	423
	424	!$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
	425	do l=1,PtrHallo%NbLevel
	426	!cdir NODEP
	427	do ij=0,Nb
	428	Buffer(Pos+ij)=Field(Offset+ij,l)
	429	enddo
	430
	431	Pos=Pos+Nb+1
	432	enddo
	433	!$OMP END DO NOWAIT
	434	enddo
	435
	436	!$OMP CRITICAL (MPI)
	437
	438	#ifdef CPP_MPI
	439	call MPI_ISSEND(Buffer(req%Pos),SizeBuffer,MPI_REAL_LMDZ,rank,a_request%tag+1000*omp_rank, &
	440	COMM_LMDZ,Req%MSG_Request,ierr)
	441	#endif
	442	IF (.NOT.using_mpi) THEN
	443	PRINT *,'Erreur, echange MPI en mode sequentiel !!!'
	444	STOP
	445	ENDIF
	446	! PRINT *,"-------------------------------------------------------------------"
	447	! PRINT *,"Process de rang",mpi_rank,"Task : ",omp_rank,"--->"
	448	! PRINT ,"Requete envoye au proc :",rank,"tag :",a_request%tag+1000omp_rank
	449	! PRINT *,"Taille du message :",SizeBuffer,"requete no :",Req%MSG_Request
	450	! PRINT *,"-------------------------------------------------------------------"
	451	!$OMP END CRITICAL (MPI)
	452	endif
	453
	454	enddo
	455
	456
	457	do rank=0,MPI_SIZE-1
	458
	459	Req=>a_Request%RequestRecv(rank)
	460	SizeBuffer=0
	461
	462	do i=1,Req%NbRequest
	463	PtrHallo=>Req%Hallo(i)
	464
	465	!$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
	466	DO l=1,PtrHallo%NbLevel
	467	SizeBuffer=SizeBuffer+PtrHallo%size*iip1
	468	ENDDO
	469	!$OMP ENDDO NOWAIT
	470	enddo
	471
	472	if (SizeBuffer>0) then
	473
	474	call allocate_buffer(SizeBuffer,Req%Index,Req%Pos)
	475	!$OMP CRITICAL (MPI)
	476
	477	#ifdef CPP_MPI
	478	call MPI_IRECV(Buffer(Req%Pos),SizeBuffer,MPI_REAL_LMDZ,rank,a_request%tag+1000*omp_rank, &
	479	COMM_LMDZ,Req%MSG_Request,ierr)
	480	#endif
	481	IF (.NOT.using_mpi) THEN
	482	PRINT *,'Erreur, echange MPI en mode sequentiel !!!'
	483	STOP
	484	ENDIF
	485
	486	! PRINT *,"-------------------------------------------------------------------"
	487	! PRINT *,"Process de rang",mpi_rank,"Task : ",omp_rank,"--->"
	488	! PRINT ,"Requete en attente du proc :",rank,"tag :",a_request%tag+1000omp_rank
	489	! PRINT *,"Taille du message :",SizeBuffer,"requete no :",Req%MSG_Request
	490	! PRINT *,"-------------------------------------------------------------------"
	491
	492	!$OMP END CRITICAL (MPI)
	493	endif
	494
	495	enddo
	496
	497	end subroutine SendRequest
	498
	499	subroutine WaitRequest(a_Request)
	500	implicit none
	501
	502	#include "dimensions.h"
	503	#include "paramet.h"
	504	#ifdef CPP_MPI
	505	include 'mpif.h'
	506	#endif
	507
	508	type(request),target :: a_request
	509	type(request_SR),pointer :: Req
	510	type(Hallo),pointer :: PtrHallo
	511	integer, dimension(2*mpi_size) :: TabRequest
	512	#ifdef CPP_MPI
	513	integer, dimension(MPI_STATUS_SIZE,2*mpi_size) :: TabStatus
	514	#else
	515	integer, dimension(1,2*mpi_size) :: TabStatus
	516	#endif
	517	integer :: NbRequest
	518	integer :: i,rank,pos,ij,l,ierr
	519	integer :: offset
	520	integer :: Nb
	521
	522
	523	NbRequest=0
	524	do rank=0,MPI_SIZE-1
	525	Req=>a_request%RequestSend(rank)
	526	if (Req%NbRequest>0) then
	527	NbRequest=NbRequest+1
	528	TabRequest(NbRequest)=Req%MSG_Request
	529	endif
	530	enddo
	531
	532	do rank=0,MPI_SIZE-1
	533	Req=>a_request%RequestRecv(rank)
	534	if (Req%NbRequest>0) then
	535	NbRequest=NbRequest+1
	536	TabRequest(NbRequest)=Req%MSG_Request
	537	endif
	538	enddo
	539
	540	if (NbRequest>0) then
	541	!$OMP CRITICAL (MPI)
	542	! PRINT *,"-------------------------------------------------------------------"
	543	! PRINT *,"Process de rang",mpi_rank,"Task : ",omp_rank,"--->",NbRequest,"en attente"
	544	! PRINT *,"No des requetes :",TabRequest(1:NbRequest)
	545	#ifdef CPP_MPI
	546	call MPI_WAITALL(NbRequest,TabRequest,TabStatus,ierr)
	547	#endif
	548	! PRINT *,"Process de rang",mpi_rank,"Task : ",omp_rank,"--->",NbRequest,"complete"
	549	! PRINT *,"-------------------------------------------------------------------"
	550	!$OMP END CRITICAL (MPI)
	551	endif
	552	do rank=0,MPI_Size-1
	553	Req=>a_request%RequestRecv(rank)
	554	if (Req%NbRequest>0) then
	555	Pos=Req%Pos
	556	do i=1,Req%NbRequest
	557	PtrHallo=>Req%Hallo(i)
	558	offset=(PtrHallo%offset-1)*iip1+1
	559	Nb=iip1*PtrHallo%size-1
	560
	561	!$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
	562	do l=1,PtrHallo%NbLevel
	563	!cdir NODEP
	564	do ij=0,Nb
	565	PtrHallo%Field(offset+ij,l)=Buffer(Pos+ij)
	566	enddo
	567
	568	Pos=Pos+Nb+1
	569	enddo
	570	!$OMP ENDDO NOWAIT
	571	enddo
	572	endif
	573	enddo
	574
	575	do rank=0,MPI_SIZE-1
	576	Req=>a_request%RequestSend(rank)
	577	if (Req%NbRequest>0) then
	578	call deallocate_buffer(Req%Index)
	579	Req%NbRequest=0
	580	endif
	581	enddo
	582
	583	do rank=0,MPI_SIZE-1
	584	Req=>a_request%RequestRecv(rank)
	585	if (Req%NbRequest>0) then
	586	call deallocate_buffer(Req%Index)
	587	Req%NbRequest=0
	588	endif
	589	enddo
	590
	591	a_request%tag=1
	592	end subroutine WaitRequest
	593
	594	subroutine WaitSendRequest(a_Request)
	595	implicit none
	596
	597	#include "dimensions.h"
	598	#include "paramet.h"
	599	#ifdef CPP_MPI
	600	include 'mpif.h'
	601	#endif
	602	type(request),target :: a_request
	603	type(request_SR),pointer :: Req
	604	type(Hallo),pointer :: PtrHallo
	605	integer, dimension(mpi_size) :: TabRequest
	606	#ifdef CPP_MPI
	607	integer, dimension(MPI_STATUS_SIZE,mpi_size) :: TabStatus
	608	#else
	609	integer, dimension(1,mpi_size) :: TabStatus
	610	#endif
	611	integer :: NbRequest
	612	integer :: i,rank,pos,ij,l,ierr
	613	integer :: offset
	614
	615
	616	NbRequest=0
	617	do rank=0,MPI_SIZE-1
	618	Req=>a_request%RequestSend(rank)
	619	if (Req%NbRequest>0) then
	620	NbRequest=NbRequest+1
	621	TabRequest(NbRequest)=Req%MSG_Request
	622	endif
	623	enddo
	624
	625
	626	if (NbRequest>0) THEN
	627	!$OMP CRITICAL (MPI)
	628	! PRINT *,"-------------------------------------------------------------------"
	629	! PRINT *,"Process de rang",mpi_rank,"Task : ",omp_rank,"--->",NbRequest,"en attente"
	630	! PRINT *,"No des requetes :",TabRequest(1:NbRequest)
	631	#ifdef CPP_MPI
	632	call MPI_WAITALL(NbRequest,TabRequest,TabStatus,ierr)
	633	#endif
	634	! PRINT *,"Process de rang",mpi_rank,"Task : ",omp_rank,"--->",NbRequest,"complete"
	635	! PRINT *,"-------------------------------------------------------------------"
	636
	637	!$OMP END CRITICAL (MPI)
	638	endif
	639
	640	do rank=0,MPI_SIZE-1
	641	Req=>a_request%RequestSend(rank)
	642	if (Req%NbRequest>0) then
	643	call deallocate_buffer(Req%Index)
	644	Req%NbRequest=0
	645	endif
	646	enddo
	647
	648	a_request%tag=1
	649	end subroutine WaitSendRequest
	650
	651	subroutine WaitRecvRequest(a_Request)
	652	implicit none
	653
	654	#include "dimensions.h"
	655	#include "paramet.h"
	656	#ifdef CPP_MPI
	657	include 'mpif.h'
	658	#endif
	659
	660	type(request),target :: a_request
	661	type(request_SR),pointer :: Req
	662	type(Hallo),pointer :: PtrHallo
	663	integer, dimension(mpi_size) :: TabRequest
	664	#ifdef CPP_MPI
	665	integer, dimension(MPI_STATUS_SIZE,mpi_size) :: TabStatus
	666	#else
	667	integer, dimension(1,mpi_size) :: TabStatus
	668	#endif
	669	integer :: NbRequest
	670	integer :: i,rank,pos,ij,l,ierr
	671	integer :: offset,Nb
	672
	673
	674	NbRequest=0
	675
	676	do rank=0,MPI_SIZE-1
	677	Req=>a_request%RequestRecv(rank)
	678	if (Req%NbRequest>0) then
	679	NbRequest=NbRequest+1
	680	TabRequest(NbRequest)=Req%MSG_Request
	681	endif
	682	enddo
	683
	684
	685	if (NbRequest>0) then
	686	!$OMP CRITICAL (MPI)
	687	! PRINT *,"-------------------------------------------------------------------"
	688	! PRINT *,"Process de rang",mpi_rank,"Task : ",omp_rank,"--->",NbRequest,"en attente"
	689	! PRINT *,"No des requetes :",TabRequest(1:NbRequest)
	690	#ifdef CPP_MPI
	691	call MPI_WAITALL(NbRequest,TabRequest,TabStatus,ierr)
	692	#endif
	693	! PRINT *,"Process de rang",mpi_rank,"Task : ",omp_rank,"--->",NbRequest,"complete"
	694	! PRINT *,"-------------------------------------------------------------------"
	695	!$OMP END CRITICAL (MPI)
	696	endif
	697
	698	do rank=0,MPI_Size-1
	699	Req=>a_request%RequestRecv(rank)
	700	if (Req%NbRequest>0) then
	701	Pos=Req%Pos
	702	do i=1,Req%NbRequest
	703	PtrHallo=>Req%Hallo(i)
	704	offset=(PtrHallo%offset-1)*iip1+1
	705	Nb=iip1*PtrHallo%size-1
	706	!$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
	707	do l=1,PtrHallo%NbLevel
	708	!cdir NODEP
	709	do ij=0,Nb
	710	PtrHallo%Field(offset+ij,l)=Buffer(Pos+ij)
	711	enddo
	712	Pos=Pos+Nb+1
	713	enddo
	714	!$OMP END DO NOWAIT
	715	enddo
	716	endif
	717	enddo
	718
	719
	720	do rank=0,MPI_SIZE-1
	721	Req=>a_request%RequestRecv(rank)
	722	if (Req%NbRequest>0) then
	723	call deallocate_buffer(Req%Index)
	724	Req%NbRequest=0
	725	endif
	726	enddo
	727
	728	a_request%tag=1
	729	end subroutine WaitRecvRequest
	730
	731
	732
	733	subroutine CopyField(FieldS,FieldR,ij,ll,jj_Nb_New)
	734
	735	implicit none
	736	#include "dimensions.h"
	737	#include "paramet.h"
	738
	739	INTEGER :: ij,ll,l
	740	REAL, dimension(ij,ll) :: FieldS
	741	REAL, dimension(ij,ll) :: FieldR
	742	integer,dimension(0:MPI_Size-1) :: jj_Nb_New
	743	integer,dimension(0:MPI_Size-1) :: jj_Begin_New,jj_End_New
	744
	745	integer ::i,jje,jjb,ijb,ije
	746
	747	jj_begin_New(0)=1
	748	jj_End_New(0)=jj_Nb_New(0)
	749	do i=1,MPI_Size-1
	750	jj_begin_New(i)=jj_end_New(i-1)+1
	751	jj_end_New(i)=jj_begin_new(i)+jj_Nb_New(i)-1
	752	enddo
	753
	754	jjb=max(jj_begin,jj_begin_new(MPI_Rank))
	755	jje=min(jj_end,jj_end_new(MPI_Rank))
	756	if (ij==ip1jm) jje=min(jje,jjm)
	757
	758	if (jje >= jjb) then
	759	ijb=(jjb-1)*iip1+1
	760	ije=jje*iip1
	761
	762	!$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
	763	do l=1,ll
	764	FieldR(ijb:ije,l)=FieldS(ijb:ije,l)
	765	enddo
	766	!$OMP ENDDO NOWAIT
	767	endif
	768
	769
	770	end subroutine CopyField
	771
	772	subroutine CopyFieldHallo(FieldS,FieldR,ij,ll,jj_Nb_New,Up,Down)
	773
	774	implicit none
	775	#include "dimensions.h"
	776	#include "paramet.h"
	777
	778	INTEGER :: ij,ll,Up,Down
	779	REAL, dimension(ij,ll) :: FieldS
	780	REAL, dimension(ij,ll) :: FieldR
	781	integer,dimension(0:MPI_Size-1) :: jj_Nb_New
	782	integer,dimension(0:MPI_Size-1) :: jj_Begin_New,jj_End_New
	783
	784	integer ::i,jje,jjb,ijb,ije,l
	785
	786
	787	jj_begin_New(0)=1
	788	jj_End_New(0)=jj_Nb_New(0)
	789	do i=1,MPI_Size-1
	790	jj_begin_New(i)=jj_end_New(i-1)+1
	791	jj_end_New(i)=jj_begin_new(i)+jj_Nb_New(i)-1
	792	enddo
	793
	794
	795	jjb=max(jj_begin,jj_begin_new(MPI_Rank)-Up)
	796	jje=min(jj_end,jj_end_new(MPI_Rank)+Down)
	797	if (ij==ip1jm) jje=min(jje,jjm)
	798
	799
	800	if (jje >= jjb) then
	801	ijb=(jjb-1)*iip1+1
	802	ije=jje*iip1
	803
	804	!$OMP DO SCHEDULE(STATIC,OMP_CHUNK)
	805	do l=1,ll
	806	FieldR(ijb:ije,l)=FieldS(ijb:ije,l)
	807	enddo
	808	!$OMP ENDDO NOWAIT
	809
	810	endif
	811	end subroutine CopyFieldHallo
	812
	813	end module mod_Hallo
	814

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

Download in other formats:

Original Format