Changeset 4445 for LMDZ6/branches

Timestamp:

Feb 28, 2023, 7:32:27 PM (15 months ago)

Author:

Laurent Fairhead

Message:

Updated routine for GPU port

File:

• LMDZ6/branches/Portage_acc/libf/dyn3d/replay3d.F90 (modified) (3 diffs)

LMDZ6/branches/Portage_acc/libf/dyn3d/replay3d.F90

@@ -21 +21 @@
         grossismx, grossismy, dzoomx, dzoomy,taux,tauy
   USE mod_const_mpi, ONLY: comm_lmdz
-
+#ifdef _OPENACC
+    USE openacc
+#endif
 
 
@@ -88 +90 @@
   INTEGER :: an, mois, jour
   REAL :: heure
+  REAL(kind=8) :: elapsed_time, mean_elapsed=0.  ! real time in seconds
+
 
   !   Initialisations:
   !   ----------------
-
+#ifdef _OPENACC
+    INTEGER :: numdev
+
+    !$acc init
+
+    ! get the number of NVIDIA devices on this node
+    numdev = acc_get_num_devices(ACC_DEVICE_NVIDIA)
+    PRINT *, 'init_physics : number of available GPU devices', numdev
+    IF (numdev < 1) STOP "Error: there are no devices available on this host. ABORTING"
+    CALL acc_init(ACC_DEVICE_NVIDIA)
+    numdev = acc_get_device_num(ACC_DEVICE_NVIDIA)
+    PRINT *, 'init_physics : GPU device id = ', numdev
+#endif
 
   !----------------------------------------------------------------------
@@ -187 +203 @@
 ! Boucle en temps sur l'appel à la parametrisation
 !---------------------------------------------------------------------
+!      ntime = 97
+      write(6,*)'number of loops to do = '
+      read(5,*) ntime
       DO it=1,ntime
-         print*,'Pas de temps ',it,klon,klev
-         call call_param_replay(klon,klev)
+!         print*,'Pas de temps ',it,klon,klev
+         call call_param_replay(klon,klev,elapsed_time)
+         mean_elapsed=mean_elapsed+elapsed_time/real(ntime)
       ENDDO
-
+      write(6,*)'Mean elapsed time = ',mean_elapsed
 
 END PROGRAM replay3d