damping gpu

CliMA · Jan 1, 2024 · 1f9ac77 · 1f9ac77
1 parent fa86172
commit 1f9ac77
Show file tree

Hide file tree

Showing 13 changed files with 780 additions and 12 deletions.
diff --git a/Examples/testNHSphere.jl b/Examples/testNHSphere.jl
@@ -25,9 +25,9 @@ RhoRPos = parsed_args["RhoRPos"]
 HorLimit = parsed_args["HorLimit"]
 Upwind = parsed_args["Upwind"]
 Damping = parsed_args["Damping"]
-Geos = parsed_args["Geos"]
 Relax = parsed_args["Relax"]
 StrideDamp = parsed_args["StrideDamp"]
+Geos = parsed_args["Geos"]
 Coriolis = parsed_args["Coriolis"]
 CoriolisType = parsed_args["CoriolisType"]
 Buoyancy = parsed_args["Buoyancy"]
@@ -248,6 +248,11 @@ if Microphysics
     @show "False Type Microphysics"  
   end
 end  
+# Damping
+if Damping
+  Damp = GPU.DampingW()(H,StrideDamp,Relax,Model.wPos)
+  Model.Damp = Damp
+end
 
 # Surface flux
 if Model.SurfaceFlux || Model.VerticalDiffusion

diff --git a/GlobInd b/GlobInd
diff --git a/Jobs/NHSphere/JobNHHeldSuarezMoistSphere b/Jobs/NHSphere/JobNHHeldSuarezMoistSphere
@@ -1,4 +1,4 @@
-mpirun -n 1 julia --project Examples/testNHSphere.jl \
+mpirun -n 6 julia --project Examples/testNHSphere.jl \
   --Problem="HeldSuarezMoistSphere" \
   --Device="CPU" \
   --GPUType="Metal" \
@@ -23,6 +23,9 @@ mpirun -n 1 julia --project Examples/testNHSphere.jl \
   --Microphysics=true \
   --TypeMicrophysics="SimpleMicrophysics" \
   --Buoyancy=true \
+  --Damping=true \
+  --StrideDamp=10000 \
+  --Relax=1.0e-3 \
   --Decomp="EqualArea" \
   --SimSeconds=0 \
   --SimDays=10 \