Changeset 724 for codes/icosagcm/devel/src/kernels_hex/compute_NH_geopot.k90

Timestamp:

08/09/18 13:55:51 (6 years ago)

Author:

dubos

Message:

devel : updated generated kernels ; added missing SIMD directives

File:

• codes/icosagcm/devel/src/kernels_hex/compute_NH_geopot.k90 (modified) (10 diffs)

codes/icosagcm/devel/src/kernels_hex/compute_NH_geopot.k90

@@ -9 +9 @@
    ! compute Phi_star
    DO l = 1,llm+1
+      !DIR$ SIMD
       DO ij=ij_omp_begin_ext,ij_omp_end_ext
          Phi_star_il(ij,l) = Phi_il(ij,l) + tau*g2*(W_il(ij,l)/m_il(ij,l)-tau)
@@ -19 +20 @@
       CASE(thermo_theta)
          DO l = 1,llm
+            !DIR$ SIMD
             DO ij=ij_omp_begin_ext,ij_omp_end_ext
                rho_ij = (g*m_ik(ij,l))/(Phi_il(ij,l+1)-Phi_il(ij,l))
@@ -29 +31 @@
       CASE(thermo_entropy)
          DO l = 1,llm
+            !DIR$ SIMD
             DO ij=ij_omp_begin_ext,ij_omp_end_ext
                rho_ij = (g*m_ik(ij,l))/(Phi_il(ij,l+1)-Phi_il(ij,l))
@@ -44 +47 @@
       ! Solve -A(l-1)x(l-1) + B(l)x(l) - A(l)x(l+1) = R(l) using Thomas algorithm
       ! Compute residual R_il and B_il
+      !DIR$ SIMD
       DO ij=ij_omp_begin_ext,ij_omp_end_ext
          ! bottom interface l=1
@@ -52 +56 @@
       END DO
       DO l = 2,llm
+         !DIR$ SIMD
          DO ij=ij_omp_begin_ext,ij_omp_end_ext
             ! inner interfaces
@@ -63 +68 @@
          END DO
       END DO
+      !DIR$ SIMD
       DO ij=ij_omp_begin_ext,ij_omp_end_ext
          ! top interface l=llm+1
@@ -74 +80 @@
       ! C(0)=0, C(l) = -A(l) / (B(l)+A(l-1)C(l-1)),
       ! D(0)=0, D(l) = (R(l)+A(l-1)D(l-1)) / (B(l)+A(l-1)C(l-1))
+      !DIR$ SIMD
       DO ij=ij_omp_begin_ext,ij_omp_end_ext
          X_ij = 1./B_il(ij,1)
@@ -80 +87 @@
       END DO
       DO l = 2,llm
+         !DIR$ SIMD
          DO ij=ij_omp_begin_ext,ij_omp_end_ext
             X_ij = 1./( B_il(ij,l) + A_ik(ij,l-1)*C_ik(ij,l-1) )
@@ -86 +94 @@
          END DO
       END DO
+      !DIR$ SIMD
       DO ij=ij_omp_begin_ext,ij_omp_end_ext
          X_ij = 1./( B_il(ij,llm+1) + A_ik(ij,llm+1 -1)*C_ik(ij,llm+1 -1) )
@@ -97 +106 @@
       END DO
       DO l = llm,1,-1
+         !DIR$ SIMD
          DO ij=ij_omp_begin_ext,ij_omp_end_ext
             ! Back substitution at lower interfaces