Changeset 3634 for palm/trunk/SOURCE/tridia_solver_mod.f90

Timestamp:

Dec 18, 2018 12:31:28 PM (5 years ago)

Author:

knoop

Message:

OpenACC port for SPEC

File:

• palm/trunk/SOURCE/tridia_solver_mod.f90 (modified) (8 diffs)

palm/trunk/SOURCE/tridia_solver_mod.f90

@@ -25 +25 @@
 ! -----------------
 ! $Id$
+! OpenACC port for SPEC
+! 
+! 3274 2018-09-24 15:42:55Z knoop
 ! Modularization of all bulk cloud physics code components
 ! 
@@ -150 +153 @@
 
        USE arrays_3d,                                                          &
-           ONLY:  ddzu_pres, ddzw, rho_air_zw
+           ONLY:  ddzu_pres, ddzw, rho_air_zw, tri
 
        IMPLICIT NONE
@@ -169 +172 @@
        CALL maketri
        CALL split
+
+       !$ACC ENTER DATA &
+       !$ACC COPYIN(ddzuw(0:nz-1,1:3)) &
+       !$ACC COPYIN(tri(nxl_z:nxr_z,nys_z:nyn_z,0:nz-1,1:2))
 
     END SUBROUTINE tridia_init
@@ -290 +297 @@
 
           REAL(wp), DIMENSION(nxl_z:nxr_z,nys_z:nyn_z,0:nz-1)   ::  ar1 !< 
+          !$ACC DECLARE CREATE(ar1)
 
 !
 !--       Forward substitution
+          !$ACC PARALLEL PRESENT(ar, ar1, tri) PRIVATE(i,j,k)
           DO  k = 0, nz - 1
+             !$ACC LOOP COLLAPSE(2)
              DO  j = nys_z, nyn_z
                 DO  i = nxl_z, nxr_z
@@ -306 +316 @@
              ENDDO
           ENDDO
+          !$ACC END PARALLEL
 
 !
@@ -312 +323 @@
 !--       by zero appearing if the pressure bc is set to neumann at the top of
 !--       the model domain.
+          !$ACC PARALLEL PRESENT(ar, ar1, ddzuw, tri) PRIVATE(i,j,k)
           DO  k = nz-1, 0, -1
+             !$ACC LOOP COLLAPSE(2)
              DO  j = nys_z, nyn_z
                 DO  i = nxl_z, nxr_z
@@ -325 +338 @@
              ENDDO
           ENDDO
+          !$ACC END PARALLEL
 
 !
@@ -332 +346 @@
           IF ( ibc_p_b == 1  .AND.  ibc_p_t == 1 )  THEN
              IF ( nys_z == 0  .AND.  nxl_z == 0 )  THEN
+                !$ACC PARALLEL LOOP PRESENT(ar)
                 DO  k = 1, nz
                    ar(nxl_z,nys_z,k) = 0.0_wp