Assert atmos to not have UnionAll types

src/ClimaAtmos.jl

@@ -6,6 +6,7 @@ include(joinpath("parameters", "Parameters.jl"))
 import .Parameters as CAP
 
 include(joinpath("utils", "abbreviations.jl"))
+include(joinpath("utils", "gpu_compat.jl"))
 include(joinpath("utils", "common_spaces.jl"))
 include(joinpath("solver", "types.jl"))
 include(joinpath("solver", "cli_options.jl"))

src/solver/type_getters.jl

@@ -78,6 +78,7 @@ function get_atmos(config::AtmosConfig, params)
         surface_model = get_surface_model(parsed_args),
         numerics = get_numerics(parsed_args),
     )
+    @assert !@any_reltype(atmos, (UnionAll,))
 
     @info "AtmosModel: \n$(summary(atmos))"
     return atmos

src/utils/gpu_compat.jl

@@ -0,0 +1,30 @@
+"""
+    @any_reltype(::Any, t::Tuple, warn=true)
+
+Returns a Bool (and prints warnings) if the given
+data structure has an instance of any types in `t`.
+"""
+function any_reltype(found, obj, name, ets, pc = (); warn = true)
+    for pn in propertynames(obj)
+        prop = getproperty(obj, pn)
+        pc_full = (pc..., ".", pn)
+        pc_string = name * string(join(pc_full))
+        for et in ets
+            if prop isa et
+                warn && @warn "$pc_string::$(typeof(prop)) is a DataType"
+                found = true
+            end
+        end
+        found = found || any_reltype(found, prop, name, ets, pc_full; warn)
+    end
+    return found
+end
+macro any_reltype(obj, ets, warn = true)
+    return :(any_reltype(
+        false,
+        $(esc(obj)),
+        $(string(obj)),
+        $(esc(ets));
+        warn = $(esc(warn)),
+    ))
+end