wip

Author: nefrathenrici

.buildkite/Manifest-v1.11.toml

@@ -2024,6 +2024,12 @@ version = "0.4.34"
     ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
     SparseConnectivityTracer = "9f842d2f-2579-4b1d-911e-f412cf18a3f5"
 
+[[deps.PrecompileCI]]
+deps = ["ClimaAtmos", "ClimaComms", "ClimaCore", "ClimaParams", "Logging", "PrecompileTools"]
+path = "PrecompileCI"
+uuid = "76d61242-8ec2-4c91-8455-3234246697a2"
+version = "0.1.0"
+
 [[deps.PrecompileTools]]
 deps = ["Preferences"]
 git-tree-sha1 = "5aa36f7049a63a1528fe8f7c3f2113413ffd4e1f"

.buildkite/PrecompileCI/src/PrecompileCI.jl

@@ -6,42 +6,48 @@ import ClimaComms
 import ClimaCore: InputOutput, Meshes, Spaces, Quadratures
 import ClimaParams
 
-# @compile_workload begin
-#     with_logger(NullLogger()) do
-#         FT = Float32 # Float64?
-#         h_elem = 6 # 16, 30?
-#         z_elem = 10 # 30, 31, 63?
-#         x_elem = y_elem = 2
-#         x_max = y_max = 1e8
-#         z_max = FT(30000.0)
-#         dz_bottom = FT(500) # other values?
-#         z_stretch = Meshes.HyperbolicTangentStretching(dz_bottom) # Meshes.Uniform()
-#         bubble = true # false
-#         parsed_args =
-#             Dict{String, Any}("topography" => "NoWarp", "topo_smoothing" => false)
-#         comms_ctx = ClimaComms.context(ClimaComms.CPUSingleThreaded())
-#         deep = false
+@compile_workload begin
+    with_logger(NullLogger()) do
+        FT = Float32 # Float64?
+        h_elem = 6 # 16, 30?
+        z_elem = 10 # 30, 31, 63?
+        x_elem = y_elem = 2
+        x_max = y_max = 1e8
+        z_max = FT(30000.0)
+        dz_bottom = FT(500) # other values?
+        z_stretch = true # Meshes.Uniform()
+        bubble = true # false
+        nh_poly = 3 # GLL{4} = nh_poly + 1
+        context = ClimaComms.context(ClimaComms.CPUSingleThreaded())
+        deep_atmosphere = false
+        topography = CA.NoTopography()
+        params = CA.ClimaAtmosParameters(FT)
+        radius = CA.Parameters.planet_radius(params)
 
-#         # constants
-#         quad = Quadratures.GLL{4}()
-#         params = CA.ClimaAtmosParameters(FT)
-#         radius = CA.Parameters.planet_radius(params)
+        sphere_grid = CA.SphereGrid(
+            FT;
+            context, radius, h_elem, nh_poly, z_elem, z_max, z_stretch, 
+            dz_bottom, bubble, deep_atmosphere, 
+            topography,
+        )
+        box_grid = CA.BoxGrid(
+            FT; context, x_elem, x_max, y_elem, y_max, z_elem, z_max, nh_poly, 
+            z_stretch, dz_bottom, bubble, deep_atmosphere, 
+            periodic_x = true, periodic_y = true, topography,
+        )
+        plane_grid = CA.PlaneGrid(
+            FT; context, x_elem, x_max, z_elem, z_max, nh_poly, z_stretch, 
+            dz_bottom, bubble, deep_atmosphere, periodic_x = true, 
+            topography,
+        )
+        column_grid = CA.ColGrid(
+            FT; context, z_elem, z_max, z_stretch, dz_bottom,
+        )
 
-#         # Sphere
-#         horz_mesh = CA.cubed_sphere_mesh(; radius, h_elem)
-#         h_space = CA.make_horizontal_space(horz_mesh, quad, comms_ctx, bubble)
-#         CA.make_hybrid_spaces(h_space, z_max, z_elem, z_stretch; parsed_args)
+        for grid in [sphere_grid, box_grid, plane_grid, column_grid]
+            CA.get_spaces(grid, context)
+        end
+    end
+end
 
-#         # box
-#         horizontal_mesh = CA.periodic_rectangle_mesh(; x_max, y_max, x_elem, y_elem)
-#         h_space = CA.make_horizontal_space(horizontal_mesh, quad, comms_ctx, bubble)
-#         # This is broken
-#         # CA.make_hybrid_spaces(h_space, z_max, z_elem, z_stretch; parsed_args)
-
-#         # plane
-#         horizontal_mesh = CA.periodic_line_mesh(; x_max, x_elem)
-#         h_space = CA.make_horizontal_space(horizontal_mesh, quad, comms_ctx, bubble)
-#     end
-# end
-
-end # module Precompile
+end # module PrecompileCI

.buildkite/Project.toml

@@ -27,6 +27,7 @@ NCDatasets = "85f8d34a-cbdd-5861-8df4-14fed0d494ab"
 NullBroadcasts = "0d71be07-595a-4f89-9529-4065a4ab43a6"
 OrderedCollections = "bac558e1-5e72-5ebc-8fee-abe8a469f55d"
 Poppler_jll = "9c32591e-4766-534b-9725-b71a8799265b"
+PrecompileCI = "76d61242-8ec2-4c91-8455-3234246697a2"
 PrettyTables = "08abe8d2-0d0c-5749-adfa-8a2ac140af0d"
 Profile = "9abbd945-dff8-562f-b5e8-e1ebf5ef1b79"
 ProfileCanvas = "efd6af41-a80b-495e-886c-e51b0c7d77a3"

.buildkite/pipeline.yml

@@ -28,16 +28,7 @@ steps:
       - julia --check-bounds=yes -e 'using TOML; @assert TOML.parsefile(".buildkite/Manifest-v1.11.toml")["deps"]["ClimaAtmos"][1]["path"] == ".."'
 
       - echo "--- Instantiate .buildkite"
-      - julia --project=.buildkite -e '
-          using Pkg;
-          Pkg.Registry.update();
-          Pkg.add(PackageSpec(name="ClimaCore",      url="https://github.com/CliMA/ClimaCore.jl",      rev="ne/fd_grid"));
-          Pkg.add(PackageSpec(name="ClimaUtilities", url="https://github.com/CliMA/ClimaUtilities.jl", rev="main"));
-          Pkg.instantiate();
-          Pkg.precompile(; strict=true);
-          using CUDA; CUDA.precompile_runtime();
-          Pkg.status()
-        '
+      - "julia --project=.buildkite -e 'using Pkg; Pkg.instantiate(;verbose=true); Pkg.precompile(;strict=true); using CUDA; CUDA.precompile_runtime(); Pkg.status()'"
 
     agents:
       slurm_cpus_per_task: 8

Project.toml

@@ -44,7 +44,7 @@ ArgParse = "1"
 Artifacts = "1"
 AtmosphericProfilesLibrary = "0.1.7"
 ClimaComms = "0.6.9"
-ClimaCore = "0.14.37"
+ClimaCore = "0.14.43"
 ClimaDiagnostics = "0.2.12"
 ClimaInterpolations = "0.1.0"
 ClimaParams = "1.0.2"