Context Navigation

bands.F90 @ 808

Last change on this file since 808 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

Line
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
13	INTEGER,dimension(:),allocatable :: distrib_phys
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))
27	allocate(distrib_phys(0:MPI_Size-1))
28
29	end subroutine AllocateBands
30
31	subroutine Read_distrib
32	use parallel
33	implicit none
34
35	include "dimensions.h"
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
67	read (unit_number,*) j,distrib_phys(i)
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(:)
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
88
89
90	SUBROUTINE Set_Bands
91	USE parallel
92	USE mod_phys_lmdz_para, ONLY : jj_para_begin,jj_para_end
93	IMPLICIT NONE
94	INCLUDE 'dimensions.h'
95	INTEGER :: i
96
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
103	jj_Nb_physic(i)=jj_para_end(i)-jj_para_begin(i)+1
104	if (i/=0) then
105	if (jj_para_begin(i)==jj_para_end(i-1)) then
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
112	jj_Nb_physic_bis(i)=jj_para_end(i)-jj_para_begin(i)+1
113	if (i/=0) then
114	if (jj_para_begin(i)==jj_para_end(i-1)) then
115	jj_Nb_physic_bis(i)=jj_Nb_physic_bis(i)-1
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
120	endif
121	enddo
122
123	end subroutine Set_Bands
124
125
126	subroutine AdjustBands_caldyn
127	use times
128	use parallel
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
193	use parallel
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
259	use parallel
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
325	USE mod_phys_lmdz_para, only : klon_mpi_para_nb
326	USE parallel
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
350	Inc(i)=nint(klon_mpi_para_nb(i)*(medium-value(i))/value(i))
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
369	distrib_phys(i)=klon_mpi_para_nb(i)+inc(i)
370	enddo
371
372	end subroutine AdjustBands_physic
373
374	subroutine WriteBands
375	USE parallel
376	implicit none
377	include "dimensions.h"
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
413	write (unit_number,*) i,distrib_phys(i)
414	enddo
415
416	CLOSE(unit_number)
417	else
418	print *,'probleme lors de l ecriture des bandes'
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