Changeset 6748 for branches/2016/dev_r6519_HPC_4/NEMOGCM/NEMO/OPA_SRC/DIA/diawri.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/DIA/diawri.F90 (modified) (16 diffs)

branches/2016/dev_r6519_HPC_4/NEMOGCM/NEMO/OPA_SRC/DIA/diawri.F90

@@ -157 +157 @@
       CALL iom_put(  "sst", tsn(:,:,1,jp_tem) )    ! surface temperature
       IF ( iom_use("sbt") ) THEN
+!$OMP PARALLEL DO schedule(static) private(jj, ji, jkbot)
          DO jj = 1, jpj
             DO ji = 1, jpi
@@ -169 +170 @@
       CALL iom_put(  "sss", tsn(:,:,1,jp_sal) )    ! surface salinity
       IF ( iom_use("sbs") ) THEN
+!$OMP PARALLEL DO schedule(static) private(jj, ji, jkbot)
          DO jj = 1, jpj
             DO ji = 1, jpi
@@ -180 +182 @@
       IF ( iom_use("taubot") ) THEN                ! bottom stress
          z2d(:,:) = 0._wp
+!$OMP PARALLEL DO schedule(static) private(jj, ji, zztmpx,zztmpy)
          DO jj = 2, jpjm1
             DO ji = fs_2, fs_jpim1   ! vector opt.
@@ -197 +200 @@
       CALL iom_put(  "ssu", un(:,:,1)         )    ! surface i-current
       IF ( iom_use("sbu") ) THEN
+!$OMP PARALLEL DO schedule(static) private(jj, ji, jkbot)
          DO jj = 1, jpj
             DO ji = 1, jpi
@@ -209 +213 @@
       CALL iom_put(  "ssv", vn(:,:,1)         )    ! surface j-current
       IF ( iom_use("sbv") ) THEN
+!$OMP PARALLEL DO schedule(static) private(jj, ji,jkbot)
          DO jj = 1, jpj
             DO ji = 1, jpi
@@ -222 +227 @@
          ! Caution: in the VVL case, it only correponds to the baroclinic mass transport.
          z2d(:,:) = rau0 * e1e2t(:,:)
+!$OMP PARALLEL DO schedule(static) private(jk)
          DO jk = 1, jpk
-            z3d(:,:,jk) = wn(:,:,jk) * z2d(:,:)
+            z3d(:,:,jk) = wn(:,:,jk) * z2d(:,:) 
          END DO
          CALL iom_put( "w_masstr" , z3d )  
@@ -237 +243 @@
 
       IF ( iom_use("sstgrad") .OR. iom_use("sstgrad2") ) THEN
+!$OMP PARALLEL DO schedule(static) private(jj, ji, zztmp, zztmpx, zztmpy)
          DO jj = 2, jpjm1                                    ! sst gradient
             DO ji = fs_2, fs_jpim1   ! vector opt.
@@ -255 +262 @@
       IF( iom_use("heatc") ) THEN
          z2d(:,:)  = 0._wp 
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji)
          DO jk = 1, jpkm1
             DO jj = 1, jpj
@@ -267 +275 @@
       IF( iom_use("saltc") ) THEN
          z2d(:,:)  = 0._wp 
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji)
          DO jk = 1, jpkm1
             DO jj = 1, jpj
@@ -279 +288 @@
       IF ( iom_use("eken") ) THEN
          rke(:,:,jk) = 0._wp                               !      kinetic energy 
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji, zztmp, zztmpx, zztmpy)
          DO jk = 1, jpkm1
             DO jj = 2, jpjm1
@@ -304 +314 @@
       IF( iom_use("u_masstr") .OR. iom_use("u_heattr") .OR. iom_use("u_salttr") ) THEN
          z3d(:,:,jpk) = 0.e0
+!$OMP PARALLEL DO schedule(static) private(jk)
          DO jk = 1, jpkm1
             z3d(:,:,jk) = rau0 * un(:,:,jk) * e2u(:,:) * e3u_n(:,:,jk) * umask(:,:,jk)
@@ -312 +323 @@
       IF( iom_use("u_heattr") ) THEN
          z2d(:,:) = 0.e0 
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji)
          DO jk = 1, jpkm1
             DO jj = 2, jpjm1
@@ -325 +337 @@
       IF( iom_use("u_salttr") ) THEN
          z2d(:,:) = 0.e0 
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji)
          DO jk = 1, jpkm1
             DO jj = 2, jpjm1
@@ -339 +352 @@
       IF( iom_use("v_masstr") .OR. iom_use("v_heattr") .OR. iom_use("v_salttr") ) THEN
          z3d(:,:,jpk) = 0.e0
+!$OMP PARALLEL DO schedule(static) private(jk)
          DO jk = 1, jpkm1
             z3d(:,:,jk) = rau0 * vn(:,:,jk) * e1v(:,:) * e3v_n(:,:,jk) * vmask(:,:,jk)
@@ -347 +361 @@
       IF( iom_use("v_heattr") ) THEN
          z2d(:,:) = 0.e0 
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji)
          DO jk = 1, jpkm1
             DO jj = 2, jpjm1
@@ -360 +375 @@
       IF( iom_use("v_salttr") ) THEN
          z2d(:,:) = 0.e0 
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji)
          DO jk = 1, jpkm1
             DO jj = 2, jpjm1