make aerosol radiation work on GPU

CliMA · Jul 9, 2024 · 2b5c5c0 · 2b5c5c0
1 parent 98bac83
commit 2b5c5c0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
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,7 +262,11 @@ function set_aerosol_type(;
     OC2 = 0,
     _...,
 )
-    _, index = findmax((DST01, SSLT01, SO4, CB1, CB2, OC1, OC2))
+    maxval = max(DST01, SSLT01, SO4, CB1, CB2, OC1, OC2)
+    index = findfirst(
+        c -> c == maxval,
+        (DST01, SSLT01, SO4, CB1, CB2, OC1, OC2),
+    )::Int
     return index
 end
 set_aerosol_type(NT) = set_aerosol_type(; NT...)