Changeset 76 for trunk/SOURCES/relaxation_water_diffusion.f90

Timestamp:

06/29/16 16:21:13 (8 years ago)

Author:

dumas

Message:

OpenMP parallelization in conserv-mass-adv-diff_sept2009_mod.f90, diffusiv-polyn-0.6.f90, dragging_neff_slope_mod.f90, eaubasale-0.5_mod.f90 and relaxation_water_diffusion.f90

File:

• trunk/SOURCES/relaxation_water_diffusion.f90 (modified) (8 diffs)

trunk/SOURCES/relaxation_water_diffusion.f90

@@ -18 +18 @@
 
 subroutine relaxation_waterdif(NXX,NYY,DT,DX,vieuxHWATER,limit_hw,klimit,BMELT,INFILTR,PGMX,PGMY,KOND,KONDMAX,HWATER)
-    implicit none
+
+  !$ USE OMP_LIB
+
+  implicit none
 
 
@@ -57 +60 @@
 
 ! write(166,*)' entree  relaxation waterdif'
+!$OMP PARALLEL
+!$OMP WORKSHARE
     HWATER(:,:)= vieuxHWATER(:,:)
-
+!$OMP END WORKSHARE
     !   calcul de kmx et kmx a partir de KOND
     !   conductivite hyrdraulique sur les noeuds mineurs
     !   moyenne harmonique
     !   ----------------------------------------
-
+!$OMP DO
     do j=2,nyy
        do i=2,nxx
@@ -75 +80 @@
        end do
     end do
-
+!$OMP END DO
+
+!$OMP DO
     do j=2,nyy
        do i=2,nxx
@@ -86 +93 @@
        enddo
     enddo
-
+!$OMP END DO
 
     !   attribution des coefficients  arelax ....
@@ -100 +107 @@
     dtwdx2=dt/dx/dx
 
-arelax(:,:)=0.
-brelax(:,:)=0.
-crelax(:,:)=1.
-drelax(:,:)=0.
-erelax(:,:)=0.
-frelax(:,:)=limit_hw(:,:)
-
-
-
+!$OMP WORKSHARE
+  arelax(:,:)=0.
+  brelax(:,:)=0.
+  crelax(:,:)=1.
+  drelax(:,:)=0.
+  erelax(:,:)=0.
+  frelax(:,:)=limit_hw(:,:)
+!$OMP END WORKSHARE
+
+!$OMP DO
     do J=2,NYY-1
        do I=2,NXX-1
 
           if (klimit(i,j).eq.0) then
-
 ! calcul du vecteur
-
           FRELAX(I,J)= VIEUXHWATER(I,J)+(BMELT(I,J)-INFILTR)*DT
           frelax(i,j)=frelax(i,j)+(kmx(i,j)*pgmx(i,j)-kmx(i+1,j)*pgmx(i+1,j))*dtsrgdx
           frelax(i,j)=frelax(i,j)+(kmy(i,j)*pgmy(i,j)-kmy(i,j+1)*pgmy(i,j+1))*dtsrgdx
-
 ! calcul des diagonales      
           arelax(i,j)=-kmx(i,j)*dtwdx2        ! arelax : diagonale i-1,j 
@@ -131 +136 @@
           crelax(i,j)=1.+((kmx(i,j)+kmx(i+1,j))+(kmy(i,j)+kmy(i,j+1)))*dtwdx2 
                                               !crelax : diagonale i,j 
-
           else if (klimit(i,j).eq.1) then
              hwater(i,j)=limit_hw(i,j)
 !             write(6,*) i,j,hwater(i,j),crelax(i,j),frelax(i,j),arelax(i,j)
-
           endif
-
        end do
     end do
-
-
+!$OMP END DO
+!$OMP END PARALLEL
 
     ! Boucle de relaxation :
@@ -152 +154 @@
        ntour=ntour+1
 !       write(6,*) 'boucle de relaxation numero',ntour
-
+       !$OMP PARALLEL
+       !$OMP DO PRIVATE(reste)
        do j=2,NYY-1
           do i=2,NXX-1
@@ -161 +164 @@
 
              DELTAH(I,J) = RESTE/CRELAX(I,J)             
-
           end do
        end do
-
-deltah(:,:)=min(deltah(:,:),10.)
-deltah(:,:)=max(deltah(:,:),-10.)
-     
-
-
+       !$OMP END DO
+
+       !$OMP WORKSHARE
+       deltah(:,:)=min(deltah(:,:),10.)
+       deltah(:,:)=max(deltah(:,:),-10.)
+       !$OMP END WORKSHARE
        ! il faut faire le calcul suivant dans une autre boucle car RESTE est fonction
        ! de hwater sur les points voisins.
+       !$OMP DO
        do j=2,NYY-1
           do i=2,NXX-1
              HWATER(I,J) = HWATER(I,J) - DELTAH(I,J)
-
           end do
        end do
-
+       !$OMP END DO
 
        ! critere d'arret:
-
-      
-
        Delh=0
        Vh=0
-
+       
+       !$OMP DO REDUCTION(+:Delh)
        DO j=2,NYY-1
           DO i=2,NXX-1
              
              !   write(166,*) I,J,delh,deltah(i,j)
-                         Delh=Delh+deltah(i,j)**2
+             Delh=Delh+deltah(i,j)**2
              !            Vh=Vh+h(i,j)**2.
           END DO
        END DO
-
+       !$OMP END DO
+       !$OMP END PARALLEL
 !       write(6,*) delh,maxval(deltah),minval(deltah)
        !      testh=SQRT(Delh/Vh)