Changeset 1046 for codes/icosagcm/trunk/src/initial/etat0_heldsz.f90

Timestamp:

08/20/20 16:22:41 (4 years ago)

Author:

ymipsl

Message:

Introduce modification from A. Durocher github to make held&suarez testcase working on GPU

YM & AD

File:

• codes/icosagcm/trunk/src/initial/etat0_heldsz.f90 (modified) (4 diffs)

codes/icosagcm/trunk/src/initial/etat0_heldsz.f90

@@ -89 +89 @@
        theta=f_theta(ind)
        CALL compute_etat0_heldsz(theta_eq,theta)
-       CALL compute_theta2theta_rhodz(ps,theta,theta_rhodz(:,:,1),1)
+       CALL compute_theta2theta_rhodz(ps,theta,theta_rhodz(:,:,1),1, ondevice=.false.)
        IF(nqtot>0) THEN
           q=f_q(ind)
@@ -96 +96 @@
           IF(nqtot>2) q(:,:,3:)=0.
        END IF
+   
+       call update_device_field(f_theta_eq)
+       call update_device_field(f_theta)
+      !$acc enter data copyin(knewt_t(:)) async
+      !$acc enter data copyin(kfrict(:)) async
+
     ENDDO
   END SUBROUTINE etat0
@@ -236 +242 @@
     INTEGER :: i,j,l,ij
 
-    CALL compute_theta_rhodz2theta(ps,theta_rhodz,theta,1)
+    CALL compute_theta_rhodz2theta(ps,theta_rhodz,theta,1, ondevice=.TRUE.)
     DO l=ll_begin,ll_end
        DO j=jj_begin-1,jj_end+1
@@ -246 +252 @@
        ENDDO
     ENDDO
-    CALL compute_theta2theta_rhodz(ps,theta,theta_rhodz,1)
-
-    Do l=ll_begin,ll_end
+    CALL compute_theta2theta_rhodz(ps,theta,theta_rhodz,1, ondevice=.true.)
+
+    !$acc kernels default(present) async
+    DO l=ll_begin,ll_end
        u(:,l)=u(:,l)*(1.-itau_physics*dt*kfrict(l))
     END DO
+    !$acc end kernels
 
   END SUBROUTINE compute_heldsz