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

branches/2016/dev_r6519_HPC_4/NEMOGCM/NEMO/OPA_SRC/DYN/dynkeg.F90

@@ -92 +92 @@
       IF( l_trddyn ) THEN           ! Save ua and va trends
          CALL wrk_alloc( jpi,jpj,jpk,   ztrdu, ztrdv )
+!$OMP PARALLEL WORKSHARE
          ztrdu(:,:,:) = ua(:,:,:) 
          ztrdv(:,:,:) = va(:,:,:) 
+!$OMP END PARALLEL WORKSHARE
       ENDIF
       
-      zhke(:,:,jpk) = 0._wp
+         zhke(:,:,jpk) = 0._wp
       
       SELECT CASE ( kscheme )             !== Horizontal kinetic energy at T-point  ==!
       !
       CASE ( nkeg_C2 )                          !--  Standard scheme  --!
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji, zu, zv)
          DO jk = 1, jpkm1
             DO jj = 2, jpj
@@ -114 +117 @@
          !
       CASE ( nkeg_HW )                          !--  Hollingsworth scheme  --!
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji, zu, zv)
          DO jk = 1, jpkm1
             DO jj = 2, jpjm1       
@@ -134 +138 @@
       END SELECT
       !
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji)
       DO jk = 1, jpkm1                    !==  grad( KE ) added to the general momentum trends  ==!
          DO jj = 2, jpjm1
@@ -144 +149 @@
       !
       IF( l_trddyn ) THEN                 ! save the Kinetic Energy trends for diagnostic
+!$OMP PARALLEL WORKSHARE
          ztrdu(:,:,:) = ua(:,:,:) - ztrdu(:,:,:)
          ztrdv(:,:,:) = va(:,:,:) - ztrdv(:,:,:)
+!$OMP END PARALLEL WORKSHARE
          CALL trd_dyn( ztrdu, ztrdv, jpdyn_keg, kt )
          CALL wrk_dealloc( jpi,jpj,jpk,   ztrdu, ztrdv )