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

branches/2016/dev_r6519_HPC_4/NEMOGCM/NEMO/OPA_SRC/TRA/trasbc.F90

@@ -88 +88 @@
       IF( l_trdtra ) THEN                    !* Save ta and sa trends
          CALL wrk_alloc( jpi, jpj, jpk, ztrdt, ztrds ) 
+!$OMP PARALLEL WORKSHARE
          ztrdt(:,:,:) = tsa(:,:,:,jp_tem)
          ztrds(:,:,:) = tsa(:,:,:,jp_sal)
+!$OMP END PARALLEL WORKSHARE
       ENDIF
       !
 !!gm  This should be moved into sbcmod.F90 module ? (especially now that ln_traqsr is read in namsbc namelist)
       IF( .NOT.ln_traqsr ) THEN     ! no solar radiation penetration
+!$OMP PARALLEL WORKSHARE
          qns(:,:) = qns(:,:) + qsr(:,:)      ! total heat flux in qns
          qsr(:,:) = 0._wp                     ! qsr set to zero
+!$OMP END PARALLEL WORKSHARE
       ENDIF
 
@@ -111 +115 @@
          ELSE                                   ! No restart or restart not found: Euler forward time stepping
             zfact = 1._wp
+!$OMP PARALLEL WORKSHARE
             sbc_tsc_b(:,:,:) = 0._wp
+!$OMP END PARALLEL WORKSHARE
          ENDIF
       ELSE                                !* other time-steps: swap of forcing fields
          zfact = 0.5_wp
+!$OMP PARALLEL WORKSHARE
          sbc_tsc_b(:,:,:) = sbc_tsc(:,:,:)
+!$OMP END PARALLEL WORKSHARE
       ENDIF
       !                             !==  Now sbc tracer content fields  ==!
+!$OMP PARALLEL DO schedule(static) private(jj, ji)
       DO jj = 2, jpj
          DO ji = fs_2, fs_jpim1   ! vector opt.
@@ -125 +134 @@
       END DO
       IF( ln_linssh ) THEN                !* linear free surface  
+!$OMP PARALLEL DO schedule(static) private(jj, ji)
          DO jj = 2, jpj                         !==>> add concentration/dilution effect due to constant volume cell
             DO ji = fs_2, fs_jpim1   ! vector opt.
@@ -136 +146 @@
       !
       DO jn = 1, jpts               !==  update tracer trend  ==!
+!$OMP PARALLEL DO schedule(static) private(jj, ji)
          DO jj = 2, jpj
             DO ji = fs_2, fs_jpim1   ! vector opt.  
@@ -217 +228 @@
       !
       IF( ln_iscpl .AND. ln_hsb) THEN         ! input of heat and salt due to river runoff 
+!$OMP PARALLEL DO schedule(static) private(jk, jj, ji, zdep)
          DO jk = 1,jpk
             DO jj = 2, jpj 
@@ -231 +243 @@
 
       IF( l_trdtra )   THEN                      ! save the horizontal diffusive trends for further diagnostics
+!$OMP PARALLEL WORKSHARE
          ztrdt(:,:,:) = tsa(:,:,:,jp_tem) - ztrdt(:,:,:)
          ztrds(:,:,:) = tsa(:,:,:,jp_sal) - ztrds(:,:,:)
+!$OMP END PARALLEL WORKSHARE
          CALL trd_tra( kt, 'TRA', jp_tem, jptra_nsr, ztrdt )
          CALL trd_tra( kt, 'TRA', jp_sal, jptra_nsr, ztrds )