Context Navigation

bands.F90 @ 801

Last change on this file since 801 was 792, checked in by Laurent Fairhead, 17 years ago
Modifications suite a la transformation des fichiers include pour qu'ils soient compatibles a la fois au format fixe et au format libre Un bon nombre de fichier *.inc du coup disparaissent LF
Property svn:eol-style set to `native` Property svn:keywords set to `Author Date Id Revision`
File size: 11.1 KB

Rev	Line
[630]	1	module Bands
	2
	3	integer, parameter :: bands_caldyn=1
	4	integer, parameter :: bands_vanleer=2
	5	integer, parameter :: bands_dissip=3
	6
	7	INTEGER,dimension(:),allocatable :: jj_Nb_Caldyn
	8	INTEGER,dimension(:),allocatable :: jj_Nb_vanleer
	9	INTEGER,dimension(:),allocatable :: jj_Nb_vanleer2
	10	INTEGER,dimension(:),allocatable :: jj_Nb_dissip
	11	INTEGER,dimension(:),allocatable :: jj_Nb_physic
	12	INTEGER,dimension(:),allocatable :: jj_Nb_physic_bis
[774]	13	INTEGER,dimension(:),allocatable :: distrib_phys
[630]	14
	15	contains
	16
	17	subroutine AllocateBands
	18	use parallel
	19	implicit none
	20
	21	allocate(jj_Nb_Caldyn(0:MPI_Size-1))
	22	allocate(jj_Nb_vanleer(0:MPI_Size-1))
	23	allocate(jj_Nb_vanleer2(0:MPI_Size-1))
	24	allocate(jj_Nb_dissip(0:MPI_Size-1))
	25	allocate(jj_Nb_physic(0:MPI_Size-1))
	26	allocate(jj_Nb_physic_bis(0:MPI_Size-1))
[774]	27	allocate(distrib_phys(0:MPI_Size-1))
[630]	28
	29	end subroutine AllocateBands
	30
[774]	31	subroutine Read_distrib
[630]	32	use parallel
	33	implicit none
	34
[792]	35	include "dimensions.h"
[630]	36	integer :: i,j
	37	character (len=4) :: siim,sjjm,sllm,sproc
	38	character (len=255) :: filename
	39	integer :: unit_number=10
	40	integer :: ierr
	41
	42	call AllocateBands
	43	write(siim,'(i3)') iim
	44	write(sjjm,'(i3)') jjm
	45	write(sllm,'(i3)') llm
	46	write(sproc,'(i3)') mpi_size
	47	filename='Bands_'//TRIM(ADJUSTL(siim))//'x'//TRIM(ADJUSTL(sjjm))//'x'//TRIM(ADJUSTL(sllm))//'_' &
	48	//TRIM(ADJUSTL(sproc))//'prc.dat'
	49
	50	OPEN(UNIT=unit_number,FILE=trim(filename),STATUS='old',FORM='formatted',IOSTAT=ierr)
	51
	52	if (ierr==0) then
	53
	54	do i=0,mpi_size-1
	55	read (unit_number,*) j,jj_nb_caldyn(i)
	56	enddo
	57
	58	do i=0,mpi_size-1
	59	read (unit_number,*) j,jj_nb_vanleer(i)
	60	enddo
	61
	62	do i=0,mpi_size-1
	63	read (unit_number,*) j,jj_nb_dissip(i)
	64	enddo
	65
	66	do i=0,mpi_size-1
[774]	67	read (unit_number,*) j,distrib_phys(i)
[630]	68	enddo
	69
	70	CLOSE(unit_number)
	71
	72	else
	73	do i=0,mpi_size-1
	74	jj_nb_caldyn(i)=(jjm+1)/mpi_size
	75	if (i<MOD(jjm+1,mpi_size)) jj_nb_caldyn(i)=jj_nb_caldyn(i)+1
	76	enddo
	77
	78	jj_nb_vanleer(:)=jj_nb_caldyn(:)
	79	jj_nb_dissip(:)=jj_nb_caldyn(:)
[774]	80
	81	do i=0,mpi_size-1
	82	distrib_phys(i)=(iim*(jjm-1)+2)/mpi_size
	83	IF (i<MOD(iim*(jjm-1)+2,mpi_size)) distrib_phys(i)=distrib_phys(i)+1
	84	enddo
	85	endif
	86
	87	end subroutine Read_distrib
[630]	88
[774]	89
	90	SUBROUTINE Set_Bands
	91	USE parallel
	92	USE mod_phys_lmdz_para, ONLY : jj_para_begin,jj_para_end
	93	IMPLICIT NONE
[792]	94	INCLUDE 'dimensions.h'
[774]	95	INTEGER :: i
	96
[630]	97	do i=0,mpi_size-1
	98	jj_nb_vanleer2(i)=(jjm+1)/mpi_size
	99	if (i<MOD(jjm+1,mpi_size)) jj_nb_vanleer2(i)=jj_nb_vanleer2(i)+1
	100	enddo
	101
	102	do i=0,MPI_Size-1
[774]	103	jj_Nb_physic(i)=jj_para_end(i)-jj_para_begin(i)+1
[630]	104	if (i/=0) then
[774]	105	if (jj_para_begin(i)==jj_para_end(i-1)) then
[630]	106	jj_Nb_physic(i-1)=jj_Nb_physic(i-1)-1
	107	endif
	108	endif
	109	enddo
	110
	111	do i=0,MPI_Size-1
[774]	112	jj_Nb_physic_bis(i)=jj_para_end(i)-jj_para_begin(i)+1
[630]	113	if (i/=0) then
[774]	114	if (jj_para_begin(i)==jj_para_end(i-1)) then
[630]	115	jj_Nb_physic_bis(i)=jj_Nb_physic_bis(i)-1
[764]	116	else
	117	jj_Nb_physic_bis(i-1)=jj_Nb_physic_bis(i-1)+1
	118	jj_Nb_physic_bis(i)=jj_Nb_physic_bis(i)-1
	119	endif
[630]	120	endif
	121	enddo
	122
[774]	123	end subroutine Set_Bands
[630]	124
[774]	125
[630]	126	subroutine AdjustBands_caldyn
	127	use times
[774]	128	use parallel
[630]	129	implicit none
	130
	131	real :: minvalue,maxvalue
	132	integer :: min_proc,max_proc
	133	integer :: i,j
	134	real,allocatable,dimension(:) :: value
	135	integer,allocatable,dimension(:) :: index
	136	real :: tmpvalue
	137	integer :: tmpindex
	138
	139	allocate(value(0:mpi_size-1))
	140	allocate(index(0:mpi_size-1))
	141
	142
	143	call allgather_timer_average
	144
	145	do i=0,mpi_size-1
	146	value(i)=timer_average(jj_nb_caldyn(i),timer_caldyn,i)
	147	index(i)=i
	148	enddo
	149
	150	do i=0,mpi_size-2
	151	do j=i+1,mpi_size-1
	152	if (value(i)>value(j)) then
	153	tmpvalue=value(i)
	154	value(i)=value(j)
	155	value(j)=tmpvalue
	156
	157	tmpindex=index(i)
	158	index(i)=index(j)
	159	index(j)=tmpindex
	160	endif
	161	enddo
	162	enddo
	163
	164	maxvalue=value(mpi_size-1)
	165	max_proc=index(mpi_size-1)
	166
	167	do i=0,mpi_size-2
	168	minvalue=value(i)
	169	min_proc=index(i)
	170	if (jj_nb_caldyn(max_proc)>3) then
	171	if (timer_iteration(jj_nb_caldyn(min_proc)+1,timer_caldyn,min_proc)<=1 ) then
	172	jj_nb_caldyn(min_proc)=jj_nb_caldyn(min_proc)+1
	173	jj_nb_caldyn(max_proc)=jj_nb_caldyn(max_proc)-1
	174	exit
	175	else
	176	if (timer_average(jj_nb_caldyn(min_proc)+1,timer_caldyn,min_proc) &
	177	-timer_delta(jj_nb_caldyn(min_proc)+1,timer_caldyn,min_proc) < maxvalue) then
	178	jj_nb_caldyn(min_proc)=jj_nb_caldyn(min_proc)+1
	179	jj_nb_caldyn(max_proc)=jj_nb_caldyn(max_proc)-1
	180	exit
	181	endif
	182	endif
	183	endif
	184	enddo
	185
	186	deallocate(value)
	187	deallocate(index)
	188
	189	end subroutine AdjustBands_caldyn
	190
	191	subroutine AdjustBands_vanleer
	192	use times
[774]	193	use parallel
[630]	194	implicit none
	195
	196	real :: minvalue,maxvalue
	197	integer :: min_proc,max_proc
	198	integer :: i,j
	199	real,allocatable,dimension(:) :: value
	200	integer,allocatable,dimension(:) :: index
	201	real :: tmpvalue
	202	integer :: tmpindex
	203
	204	allocate(value(0:mpi_size-1))
	205	allocate(index(0:mpi_size-1))
	206
	207
	208	call allgather_timer_average
	209
	210	do i=0,mpi_size-1
	211	value(i)=timer_average(jj_nb_vanleer(i),timer_vanleer,i)
	212	index(i)=i
	213	enddo
	214
	215	do i=0,mpi_size-2
	216	do j=i+1,mpi_size-1
	217	if (value(i)>value(j)) then
	218	tmpvalue=value(i)
	219	value(i)=value(j)
	220	value(j)=tmpvalue
	221
	222	tmpindex=index(i)
	223	index(i)=index(j)
	224	index(j)=tmpindex
	225	endif
	226	enddo
	227	enddo
	228
	229	maxvalue=value(mpi_size-1)
	230	max_proc=index(mpi_size-1)
	231
	232	do i=0,mpi_size-2
	233	minvalue=value(i)
	234	min_proc=index(i)
	235
	236	if (jj_nb_vanleer(max_proc)>3) then
	237	if (timer_average(jj_nb_vanleer(min_proc)+1,timer_vanleer,min_proc)==0. .or. &
	238	timer_average(jj_nb_vanleer(max_proc)-1,timer_vanleer,max_proc)==0.) then
	239	jj_nb_vanleer(min_proc)=jj_nb_vanleer(min_proc)+1
	240	jj_nb_vanleer(max_proc)=jj_nb_vanleer(max_proc)-1
	241	exit
	242	else
	243	if (timer_average(jj_nb_vanleer(min_proc)+1,timer_vanleer,min_proc) < maxvalue) then
	244	jj_nb_vanleer(min_proc)=jj_nb_vanleer(min_proc)+1
	245	jj_nb_vanleer(max_proc)=jj_nb_vanleer(max_proc)-1
	246	exit
	247	endif
	248	endif
	249	endif
	250	enddo
	251
	252	deallocate(value)
	253	deallocate(index)
	254
	255	end subroutine AdjustBands_vanleer
	256
	257	subroutine AdjustBands_dissip
	258	use times
[774]	259	use parallel
[630]	260	implicit none
	261
	262	real :: minvalue,maxvalue
	263	integer :: min_proc,max_proc
	264	integer :: i,j
	265	real,allocatable,dimension(:) :: value
	266	integer,allocatable,dimension(:) :: index
	267	real :: tmpvalue
	268	integer :: tmpindex
	269
	270	allocate(value(0:mpi_size-1))
	271	allocate(index(0:mpi_size-1))
	272
	273
	274	call allgather_timer_average
	275
	276	do i=0,mpi_size-1
	277	value(i)=timer_average(jj_nb_dissip(i),timer_dissip,i)
	278	index(i)=i
	279	enddo
	280
	281	do i=0,mpi_size-2
	282	do j=i+1,mpi_size-1
	283	if (value(i)>value(j)) then
	284	tmpvalue=value(i)
	285	value(i)=value(j)
	286	value(j)=tmpvalue
	287
	288	tmpindex=index(i)
	289	index(i)=index(j)
	290	index(j)=tmpindex
	291	endif
	292	enddo
	293	enddo
	294
	295	maxvalue=value(mpi_size-1)
	296	max_proc=index(mpi_size-1)
	297
	298	do i=0,mpi_size-2
	299	minvalue=value(i)
	300	min_proc=index(i)
	301
	302	if (jj_nb_dissip(max_proc)>3) then
	303	if (timer_iteration(jj_nb_dissip(min_proc)+1,timer_dissip,min_proc)<=1) then
	304	jj_nb_dissip(min_proc)=jj_nb_dissip(min_proc)+1
	305	jj_nb_dissip(max_proc)=jj_nb_dissip(max_proc)-1
	306	exit
	307	else
	308	if (timer_average(jj_nb_dissip(min_proc)+1,timer_dissip,min_proc) &
	309	- timer_delta(jj_nb_dissip(min_proc)+1,timer_dissip,min_proc) < maxvalue) then
	310	jj_nb_dissip(min_proc)=jj_nb_dissip(min_proc)+1
	311	jj_nb_dissip(max_proc)=jj_nb_dissip(max_proc)-1
	312	exit
	313	endif
	314	endif
	315	endif
	316	enddo
	317
	318	deallocate(value)
	319	deallocate(index)
	320
	321	end subroutine AdjustBands_dissip
	322
	323	subroutine AdjustBands_physic
	324	use times
[774]	325	USE mod_phys_lmdz_para, only : klon_mpi_para_nb
	326	USE parallel
[630]	327	implicit none
	328
	329	integer :: i,Index
	330	real,allocatable,dimension(:) :: value
	331	integer,allocatable,dimension(:) :: Inc
	332	real :: medium
	333	integer :: NbTot,sgn
	334
	335	allocate(value(0:mpi_size-1))
	336	allocate(Inc(0:mpi_size-1))
	337
	338
	339	call allgather_timer_average
	340
	341	medium=0
	342	do i=0,mpi_size-1
	343	value(i)=timer_average(jj_nb_physic(i),timer_physic,i)
	344	medium=medium+value(i)
	345	enddo
	346
	347	medium=medium/mpi_size
	348	NbTot=0
	349	do i=0,mpi_size-1
[774]	350	Inc(i)=nint(klon_mpi_para_nb(i)*(medium-value(i))/value(i))
[630]	351	NbTot=NbTot+Inc(i)
	352	enddo
	353
	354	if (NbTot>=0) then
	355	Sgn=1
	356	else
	357	Sgn=-1
	358	NbTot=-NbTot
	359	endif
	360
	361	Index=0
	362	do i=1,NbTot
	363	Inc(Index)=Inc(Index)-Sgn
	364	Index=Index+1
	365	if (Index>mpi_size-1) Index=0
	366	enddo
	367
	368	do i=0,mpi_size-1
[774]	369	distrib_phys(i)=klon_mpi_para_nb(i)+inc(i)
[630]	370	enddo
	371
	372	end subroutine AdjustBands_physic
	373
	374	subroutine WriteBands
	375	USE parallel
	376	implicit none
[792]	377	include "dimensions.h"
[630]	378
	379	integer :: i,j
	380	character (len=4) :: siim,sjjm,sllm,sproc
	381	character (len=255) :: filename
	382	integer :: unit_number=10
	383	integer :: ierr
	384
	385	write(siim,'(i3)') iim
	386	write(sjjm,'(i3)') jjm
	387	write(sllm,'(i3)') llm
	388	write(sproc,'(i3)') mpi_size
	389
	390	filename='Bands_'//TRIM(ADJUSTL(siim))//'x'//TRIM(ADJUSTL(sjjm))//'x'//TRIM(ADJUSTL(sllm))//'_' &
	391	//TRIM(ADJUSTL(sproc))//'prc.dat'
	392
	393	OPEN(UNIT=unit_number,FILE=trim(filename),STATUS='replace',FORM='formatted',IOSTAT=ierr)
	394
	395	if (ierr==0) then
	396
	397	! write (unit_number,) ' Bandes caldyn *'
	398	do i=0,mpi_size-1
	399	write (unit_number,*) i,jj_nb_caldyn(i)
	400	enddo
	401
	402	! write (unit_number,) ' Bandes vanleer *'
	403	do i=0,mpi_size-1
	404	write (unit_number,*) i,jj_nb_vanleer(i)
	405	enddo
	406
	407	! write (unit_number,) ' Bandes dissip *'
	408	do i=0,mpi_size-1
	409	write (unit_number,*) i,jj_nb_dissip(i)
	410	enddo
	411
	412	do i=0,mpi_size-1
[774]	413	write (unit_number,*) i,distrib_phys(i)
[630]	414	enddo
	415
	416	CLOSE(unit_number)
	417	else
[764]	418	print *,'probleme lors de l ecriture des bandes'
[630]	419	endif
	420
	421	end subroutine WriteBands
	422
	423	end module Bands
	424
	425

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

Download in other formats:

Original Format