Changeset 1047

Timestamp:

08/26/20 11:34:31 (4 years ago)

Author:

ymipsl

Message:

GPU port of sponge layer

YM

Location:

codes/icosagcm/trunk/src

Files:

• dissip/sponge.f90 (modified) (4 diffs)
• time/timeloop_gcm.F90 (modified) (1 diff)

codes/icosagcm/trunk/src/dissip/sponge.f90

@@ -45 +45 @@
     ENDIF
 
-    IF (iflag_sponge > 0) THEN
-       CALL abort_acc("iflag_sponge > 0")
-    END IF
 
 !$OMP MASTER
     ALLOCATE(rdamp(llm))
     ALLOCATE(lambda(llm))
-
+    !$acc enter data create(rdamp(:),lambda(:)) async
+    
     IF (iflag_sponge == 1) THEN
 ! sponge quenching over the topmost 4 atmospheric layers
@@ -88 +86 @@
 !$OMP END MASTER
 !$OMP BARRIER
-
+    !$acc update device(rdamp(:), lambda(:)) async
+    
   END SUBROUTINE init_sponge
 
@@ -128 +127 @@
         dtheta_rhodz=f_dtheta_rhodz(ind)
 
+       !$acc parallel loop collapse(2) present(ue(:,:), due(:,:), rdamp(:), dtheta_rhodz)  async
         DO l=ll_begin,ll_end
 !$SIMD
@@ -139 +139 @@
           ENDDO
         ENDDO
+       !$acc end parallel loop
       END DO
     ELSE

codes/icosagcm/trunk/src/time/timeloop_gcm.F90

@@ -360 +360 @@
           CALL euler_scheme(.FALSE.)  ! update only u, theta
           IF (iflag_sponge > 0) THEN
-             CALL abort_acc("iflag_sponge>0")
              CALL sponge(f_u,f_du,f_theta_rhodz,f_dtheta_rhodz)
              CALL euler_scheme(.FALSE.)  ! update only u, theta