Changeset 1046 for codes/icosagcm/trunk/src/physics/physics.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/physics/physics.f90 (modified) (6 diffs)

codes/icosagcm/trunk/src/physics/physics.f90

@@ -113 +113 @@
 
 !$OMP PARALLEL
-    CALL allocate_field(f_du_phys,field_u,type_real,llm, name='du_phys')
+    CALL allocate_field(f_du_phys,field_u,type_real,llm, name='du_phys', ondevice=.TRUE.)
 
     IF(is_mpi_root) PRINT *, 'phys_type = ',phys_type
@@ -119 +119 @@
     
 !$OMP END PARALLEL
-    IF (phys_type /= phys_none) THEN
-       CALL abort_acc("physics /= 'none'")
+    IF (phys_type /= phys_none .AND. phys_type /= phys_HS94) THEN
+       CALL abort_acc("only phys_type == 'phys_none' or 'phys_HS94' supported")
     END IF
   END SUBROUTINE init_physics
@@ -133 +133 @@
        CALL swap_geometry(ind)
        du=f_du_phys(ind)
+
+       !$acc kernels default(present) async
        du(:,ll_begin:ll_end) = 0.
+       !$acc end kernels
+          
     END DO
   END SUBROUTINE zero_du_phys
@@ -141 +145 @@
     TYPE(t_field),POINTER :: f_u(:) ! velocity field before/after call to physics
     REAL(rstd), DIMENSION(:,:), POINTER :: u, du
-    INTEGER :: ind
+    INTEGER :: ind, ij
+
     DO ind=1,ndomain
        IF (.NOT. assigned_domain(ind)) CYCLE
@@ -148 +153 @@
        du=f_du_phys(ind)
        u=f_u(ind)
-       du(:,ll_begin:ll_end) = du(:,ll_begin:ll_end) + coef*u(:,ll_begin:ll_end)
+
+       !$acc parallel loop default(present) async
+       DO ij = ij_begin, ij_end
+          du(ij,ll_begin:ll_end) = du(ij,ll_begin:ll_end) + coef*u(ij,ll_begin:ll_end)
+       END DO
+
     END DO
   END SUBROUTINE add_du_phys
@@ -228 +238 @@
     USE wind_mod
     USE output_field_mod
-    CALL transfert_request(f_du_phys,req_e1_vect)
+
+    CALL transfert_request(f_du_phys,req_e1_vect)    
+    CALL update_host_field(f_du_phys)
+
     CALL un2ulonlat(f_du_phys, f_buf_ulon, f_buf_ulat, (1./(dt*itau_out)))
     CALL output_field("dulon_phys",f_buf_ulon)