Changeset 7698 for trunk/NEMOGCM/NEMO/OPA_SRC/step.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/step.F90 (modified) (4 diffs)

trunk/NEMOGCM/NEMO/OPA_SRC/step.F90

@@ -74 +74 @@
       !!              -8- Outputs and diagnostics
       !!----------------------------------------------------------------------
-      INTEGER ::   ji,jj,jk ! dummy loop indice
+      INTEGER ::   ji,jj,jk,jn ! dummy loop indice
       INTEGER ::   indic    ! error indicator if < 0
       INTEGER ::   kcall    ! optional integer argument (dom_vvl_sf_nxt)
@@ -135 +135 @@
       !
       IF( lk_zdfcst  ) THEN                                ! Constant Kz (reset avt, avm[uv] to the background value)
-         avt (:,:,:) = rn_avt0 * wmask (:,:,:)
-         avmu(:,:,:) = rn_avm0 * wumask(:,:,:)
-         avmv(:,:,:) = rn_avm0 * wvmask(:,:,:)
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji)
+         DO jk = 1, jpk
+            DO jj = 1, jpj
+               DO ji = 1, jpi
+                  avt (ji,jj,jk) = rn_avt0 * wmask (ji,jj,jk)
+                  avmu(ji,jj,jk) = rn_avm0 * wumask(ji,jj,jk)
+                  avmv(ji,jj,jk) = rn_avm0 * wvmask(ji,jj,jk)
+               END DO
+            END DO
+         END DO
       ENDIF
 
       IF( ln_rnf_mouth ) THEN                         ! increase diffusivity at rivers mouths
-         DO jk = 2, nkrnf   ;   avt(:,:,jk) = avt(:,:,jk) + 2._wp * rn_avt_rnf * rnfmsk(:,:) * tmask(:,:,jk)   ;   END DO
+!$OMP PARALLEL DO schedule(static) private(jk,jj,ji)
+         DO jk = 2, nkrnf
+            DO jj = 1, jpj
+               DO ji = 1, jpi
+                  avt(ji,jj,jk) = avt(ji,jj,jk) + 2._wp * rn_avt_rnf * rnfmsk(ji,jj) * tmask(ji,jj,jk)
+               END DO
+            END DO
+         END DO
       ENDIF
       IF( ln_zdfevd  )   CALL zdf_evd( kstp )         ! enhanced vertical eddy diffusivity
@@ -197 +211 @@
                &                                          rhd, gru , grv , grui, grvi   )  ! of t, s, rd at the first ocean level
 !!jc: fs simplification
-                            
-                         ua(:,:,:) = 0._wp            ! set dynamics trends to zero
-                         va(:,:,:) = 0._wp
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji)
+         DO jk = 1, jpk
+            DO jj = 1, jpj
+               DO ji = 1, jpi
+                  ua(ji,jj,jk) = 0._wp            ! set dynamics trends to zero
+                  va(ji,jj,jk) = 0._wp
+               END DO
+            END DO
+         END DO
 
       IF(  lk_asminc .AND. ln_asmiau .AND. ln_dyninc )   &
@@ -252 +272 @@
       ! Active tracers                              
       !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
-                         tsa(:,:,:,:) = 0._wp         ! set tracer trends to zero
+      DO jn = 1, jpts
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji)
+         DO jk = 1, jpk
+            DO jj = 1, jpj
+               DO ji = 1, jpi
+                  tsa(ji,jj,jk,jn) = 0._wp         ! set tracer trends to zero
+               END DO
+            END DO
+         END DO
+      END DO
 
       IF(  lk_asminc .AND. ln_asmiau .AND. &