source: LMDZ5/branches/LMDZ6_rc0/libf/dyn3dpar/bands.F90 @ 2687

Last change on this file since 2687 was 1910, checked in by Laurent Fairhead, 11 years ago

Merged trunk changes r1860:1909 into testing branch

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