make aerosol radiation work on GPU

CliMA · Jul 9, 2024 · b4c8b21 · b4c8b21
1 parent 1b7ecf2
commit b4c8b21
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/config/model_configs/gpu_aquaplanet_dyamond.yml b/config/model_configs/gpu_aquaplanet_dyamond.yml
@@ -18,5 +18,6 @@ surface_setup: "DefaultMoninObukhov"
 rayleigh_sponge: true
 dt: "100secs"
 t_end: "8hours"
+aerosol_radiation: true
 prescribed_aerosols: ["CB1", "CB2", "DST01", "DST02", "DST03", "DST04", "OC1", "OC2", "SO4", "SOA", "SSLT01", "SSLT02", "SSLT03", "SSLT04"]
 toml: [toml/longrun_aquaplanet_dyamond.toml]
diff --git a/src/callbacks/callbacks.jl b/src/callbacks/callbacks.jl
@@ -262,10 +262,9 @@ function set_aerosol_type(;
     OC2 = 0,
     _...,
 )
-    _, index = findmax((DST01, SSLT01, SO4, CB1, CB2, OC1, OC2))
-    return index
+    return argmax(n -> (DST01, SSLT01, SO4, CB1, CB2, OC1, OC2)[n], 1:7)
 end
-set_aerosol_type(NT) = set_aerosol_type(; NT...)
+set_aerosol_type(nt) = set_aerosol_type(; nt...)
 
 NVTX.@annotate function save_state_to_disk_func(integrator, output_dir)
     (; t, u, p) = integrator