Changeset 7698 for trunk/NEMOGCM/NEMO/LIM_SRC_3/limthd_da.F90

Timestamp:

2017-02-18T10:02:03+01:00 (7 years ago)

Author:

mocavero

Message:

update trunk with OpenMP parallelization

File:

• trunk/NEMOGCM/NEMO/LIM_SRC_3/limthd_da.F90 (modified) (3 diffs)

trunk/NEMOGCM/NEMO/LIM_SRC_3/limthd_da.F90

@@ -113 +113 @@
       zastar = 1._wp / ( 1._wp - (rn_dmin / zdmax)**(1._wp/rn_beta) )
       
+!$OMP PARALLEL
+!$OMP DO schedule(static) private(jj,ji,zdfloe,zperi,zwlat)
       DO jj = 1, jpj
          DO ji = 1, jpi
@@ -135 +137 @@
       !---------------------------------------------------------------------------------------------!
       DO jl = jpl, 1, -1
+!$OMP DO schedule(static) private(jj,ji,rswitch,zda)
          DO jj = 1, jpj
             DO ji = 1, jpi
@@ -163 +166 @@
       
       ! total concentration
-      at_i(:,:) = SUM( a_i(:,:,:), dim=3 )
-      
+!$OMP DO schedule(static) private(jj,ji)
+      DO jj = 1, jpj
+         DO ji = 1, jpi
+            at_i(ji,jj) = 0._wp
+         END DO
+      END DO
+      DO jl = 1, jpl
+!$OMP DO schedule(static) private(jj,ji)
+         DO jj = 1, jpj
+            DO ji = 1, jpi
+               at_i(ji,jj) = at_i(ji,jj) + a_i(ji,jj,jl)
+            END DO
+         END DO
+      END DO
       ! --- ensure that ht_i = 0 where a_i = 0 ---
-      WHERE( a_i == 0._wp ) ht_i = 0._wp
+      DO jl = 1, jpl
+!$OMP DO schedule(static) private(jj,ji)
+         DO jj = 1, jpj
+            DO ji = 1, jpi
+               IF(a_i(ji,jj,jl)  == 0._wp) ht_i(ji,jj,jl) = 0._wp
+            END DO
+         END DO
+      END DO
+!$OMP END PARALLEL
+
       !
       CALL wrk_dealloc( jpi,jpj, zda_tot )