Changeset 7698 for trunk/NEMOGCM/NEMO/OPA_SRC/ZDF/zdfddm.F90

Timestamp:

2017-02-18T10:02:03+01:00 (7 years ago)

Author:

mocavero

Message:

update trunk with OpenMP parallelization

File:

• trunk/NEMOGCM/NEMO/OPA_SRC/ZDF/zdfddm.F90 (modified) (7 diffs)

trunk/NEMOGCM/NEMO/OPA_SRC/ZDF/zdfddm.F90

@@ -112 +112 @@
          ! Define the mask 
          ! ---------------
+!$OMP PARALLEL
+!$OMP DO schedule(static) private(jj,ji,zrw,zaw,zbw,zdt,zds)
          DO jj = 1, jpj                                ! R=zrau = (alpha / beta) (dk[t] / dk[s])
             DO ji = 1, jpi
@@ -128 +130 @@
             END DO
          END DO
-
+!$OMP END DO NOWAIT
+
+!$OMP DO schedule(static) private(jj,ji)
          DO jj = 1, jpj                                     ! indicators:
             DO ji = 1, jpi
@@ -155 +159 @@
          END DO
          ! mask zmsk in order to have avt and avs masked
-         zmsks(:,:) = zmsks(:,:) * wmask(:,:,jk)
-
+
+!$OMP DO schedule(static) private(jj,ji)
+         DO jj = 1, jpj                               
+            DO ji = 1, jpi
+               zmsks(ji,jj) = zmsks(ji,jj) * wmask(ji,jj,jk)
+            END DO
+         END DO
 
          ! Update avt and avs
          ! ------------------
          ! Constant eddy coefficient: reset to the background value
+!$OMP DO schedule(static) private(jj,ji,zinr,zrr,zavfs,zavft,zavdt,zavds)
          DO jj = 1, jpj
             DO ji = 1, jpi
@@ -189 +199 @@
          ! --------------------------------
 !!gm to be changed following the definition of avm.
+!$OMP DO schedule(static) private(jj,ji)
          DO jj = 1, jpjm1
             DO ji = 1, fs_jpim1   ! vector opt.
@@ -199 +210 @@
             END DO
          END DO
+!$OMP END DO NOWAIT
+!$OMP END PARALLEL
          !                                                ! ===============
       END DO                                              !   End of slab
@@ -232 +245 @@
       !!----------------------------------------------------------------------
       INTEGER ::   ios   ! local integer
+      INTEGER  ::   ji, jj , jk     ! dummy loop indices
       !!
       NAMELIST/namzdf_ddm/ rn_avts, rn_hsbfr
@@ -257 +271 @@
       IF( zdf_ddm_alloc() /= 0 )   CALL ctl_stop( 'STOP', 'zdf_ddm_init : unable to allocate arrays' )
       !                               ! initialization to masked Kz
-      avs(:,:,:) = rn_avt0 * wmask(:,:,:) 
+!$OMP DO schedule(static) private(jk,jj,ji)
+      DO jk = 1, jpk                               
+         DO jj = 1, jpj                               
+            DO ji = 1, jpi
+               avs(ji,jj,jk) = rn_avt0 * wmask(ji,jj,jk)
+            END DO
+         END DO
+      END DO 
       !
    END SUBROUTINE zdf_ddm_init