Changeset 6748 for branches/2016/dev_r6519_HPC_4/NEMOGCM/NEMO/OPA_SRC/LDF/ldfslp.F90

Timestamp:

2016-06-28T11:53:56+02:00 (8 years ago)

Author:

mocavero

Message:

GYRE hybrid parallelization

File:

• branches/2016/dev_r6519_HPC_4/NEMOGCM/NEMO/OPA_SRC/LDF/ldfslp.F90 (modified) (15 diffs)

branches/2016/dev_r6519_HPC_4/NEMOGCM/NEMO/OPA_SRC/LDF/ldfslp.F90

@@ -135 +135 @@
       z1_slpmax = 1._wp / rn_slpmax
       !
+
+!$OMP PARALLEL WORKSHARE
       zww(:,:,:) = 0._wp
       zwz(:,:,:) = 0._wp
-      !
+!$OMP END PARALLEL WORKSHARE
+      !
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji)
       DO jk = 1, jpk             !==   i- & j-gradient of density   ==!
          DO jj = 1, jpjm1
@@ -147 +151 @@
       END DO
       IF( ln_zps ) THEN                           ! partial steps correction at the bottom ocean level
+!$OMP PARALLEL DO schedule(static) private(jj, ji)
          DO jj = 1, jpjm1
             DO ji = 1, jpim1
@@ -155 +160 @@
       ENDIF
       IF( ln_zps .AND. ln_isfcav ) THEN           ! partial steps correction at the bottom ocean level
+!$OMP PARALLEL DO schedule(static) private(jj, ji)
          DO jj = 1, jpjm1
             DO ji = 1, jpim1
@@ -164 +170 @@
       !
       zdzr(:,:,1) = 0._wp        !==   Local vertical density gradient at T-point   == !   (evaluated from N^2)
+!$OMP PARALLEL DO schedule(static) private(jk)
       DO jk = 2, jpkm1
          !                                ! zdzr = d/dz(prd)= - ( prd ) / grav * mk(pn2) -- at t point
@@ -182 +189 @@
       !
       IF ( ln_isfcav ) THEN
+!$OMP PARALLEL DO schedule(static) private(jj,ji)
          DO jj = 2, jpjm1
             DO ji = fs_2, fs_jpim1   ! vector opt.
@@ -191 +199 @@
          END DO
       ELSE
+!$OMP PARALLEL DO schedule(static) private(jj,ji)
          DO jj = 2, jpjm1
             DO ji = fs_2, fs_jpim1   ! vector opt.
@@ -198 +207 @@
          END DO
       END IF
-
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji, zau, zav, zbu, zbv, zfj, zfi, zdepu, zdepv)
       DO jk = 2, jpkm1                            !* Slopes at u and v points
          DO jj = 2, jpjm1
@@ -239 +248 @@
       !
       !                                            !* horizontal Shapiro filter
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji)
       DO jk = 2, jpkm1
          DO jj = 2, jpjm1, MAX(1, jpj-3)                        ! rows jj=2 and =jpjm1 only
@@ -283 +293 @@
       ! ===========================      | wslpj = mij( d/dj( prd ) / d/dz( prd )
       !
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji, zbw, zfk, zck, zbi, zbj, zai, zaj, zci, zcj)
       DO jk = 2, jpkm1
          DO jj = 2, jpjm1
@@ -321 +332 @@
       !
       !                                           !* horizontal Shapiro filter
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji, zcofw, zck)
       DO jk = 2, jpkm1
          DO jj = 2, jpjm1, MAX(1, jpj-3)                        ! rows jj=2 and =jpjm1 only
@@ -429 +441 @@
          !
          ip = jl   ;   jp = jl                ! guaranteed nonzero gradients ( absolute value larger than repsln)
+!$OMP PARALLEL DO schedule(static) private(jk,jj,ji,zdit,zdis,zdjt,zdjs,zdxrho_raw,zdyrho_raw)
          DO jk = 1, jpkm1                     ! done each pair of triad
             DO jj = 1, jpjm1                  ! NB: not masked ==>  a minimum value is set
@@ -445 +458 @@
          !
          IF( ln_zps .AND. l_grad_zps ) THEN     ! partial steps: correction of i- & j-grad on bottom
+!$OMP PARALLEL DO schedule(static) private(jj,ji,iku,zdit,zdis,zdxrho_raw,zdyrho_raw)
             DO jj = 1, jpjm1
                DO ji = 1, jpim1
@@ -676 +690 @@
       !
       !                                            !==   surface mixed layer mask   !
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji, ik)
       DO jk = 1, jpk                               ! =1 inside the mixed layer, =0 otherwise
          DO jj = 1, jpj
@@ -698 +713 @@
       !-----------------------------------------------------------------------
       !
+!$OMP PARALLEL DO schedule(static) private(jj, ji, iku, ikv, zbu, zbv, zau, zav, ik, ikm1, zbw, zci, zcj, zai, zaj, zbi, zbj) 
       DO jj = 2, jpjm1
          DO ji = 2, jpim1
@@ -791 +807 @@
          ! Direction of lateral diffusion (tracers and/or momentum)
          ! ------------------------------
-         uslp (:,:,:) = 0._wp   ;   uslpml (:,:) = 0._wp      ! set the slope to zero (even in s-coordinates)
-         vslp (:,:,:) = 0._wp   ;   vslpml (:,:) = 0._wp
-         wslpi(:,:,:) = 0._wp   ;   wslpiml(:,:) = 0._wp
-         wslpj(:,:,:) = 0._wp   ;   wslpjml(:,:) = 0._wp
-
+
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji)   
+        DO jk = 1, jpk
+           DO jj = 1, jpj
+              DO ji = 1, jpi
+                 uslp (ji,jj,jk) = 0._wp
+                 vslp (ji,jj,jk) = 0._wp
+                 wslpi(ji,jj,jk) = 0._wp
+                 wslpj(ji,jj,jk) = 0._wp
+              END DO
+           END DO
+        END DO
+!$OMP PARALLEL DO schedule(static) private(jj, ji)       
+        DO jj = 1, jpj
+            DO ji = 1, jpi
+               uslpml (ji,jj) = 0._wp
+               vslpml (ji,jj) = 0._wp
+               wslpiml(ji,jj) = 0._wp
+               wslpjml(ji,jj) = 0._wp
+             END DO
+        END DO
+         
          !!gm I no longer understand this.....
 !!gm         IF( (ln_traldf_hor .OR. ln_dynldf_hor) .AND. .NOT. (.NOT.ln_linssh .AND. ln_rstart) ) THEN