Add finalisation routine to gw_ml to destroy the net.

DataWaveProject · Aug 19, 2024 · aa106e6 · aa106e6
1 parent ab846d3
commit aa106e6
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/src/physics/cam/gw_drag.F90 b/src/physics/cam/gw_drag.F90
@@ -42,7 +42,8 @@ module gw_drag
   use gw_common,      only: GWBand
   use gw_convect,     only: BeresSourceDesc
   use gw_front,       only: CMSourceDesc
-  use gw_ml,          only: gw_drag_convect_dp_ml_init, gw_drag_convect_dp_ml
+  use gw_ml,          only: gw_drag_convect_dp_ml_init, gw_drag_convect_dp_ml_final, &
+                            gw_drag_convect_dp_ml
 
 ! Typical module header
   implicit none
@@ -1259,7 +1260,7 @@ end subroutine handle_pio_error
 subroutine gw_final()
   ! Destroy neccessary attributes if using ML scheme for convective drag
   if ((gw_convect_dp_ml == 'on') .or. (gw_convect_dp_ml == 'bothon')) then
-     call torch_delete(gw_convect_dp_nn)
+     call gw_drag_convect_dp_ml_final()
   endif
 end subroutine gw_final
 

diff --git a/src/physics/cam/gw_ml.F90 b/src/physics/cam/gw_ml.F90
@@ -153,6 +153,14 @@ subroutine gw_drag_convect_dp_ml_init(neural_net_path, norms_path)
 end subroutine gw_drag_convect_dp_ml_init
 
 
+subroutine gw_drag_convect_dp_ml_final()
+
+  ! Destroy the convective drag net
+  call torch_delete(convect_net)
+
+end subroutine gw_drag_convect_dp_ml_final
+
+
 subroutine read_norms(norms_path)
 
   use netcdf