Merge pull request #2122 from CliMA/ck/broken_gpu_test

Add high resolution space construction tests

Author: charleskawczynski

.buildkite/pipeline.yml

@@ -259,14 +259,15 @@ steps:
         command:
           - "julia --color=yes --check-bounds=yes --project=.buildkite test/Spaces/unit_spaces.jl"
           - "julia --color=yes --check-bounds=yes --project=.buildkite test/Spaces/opt_spaces.jl"
+          - "julia --color=yes --check-bounds=yes --project=.buildkite test/Spaces/unit_high_resolution_space.jl"
 
       - label: "Unit: cuda spaces"
         key: "gpu_cuda_spaces"
         command:
           - "julia --project=.buildkite -e 'using CUDA; CUDA.versioninfo()'"
           - "srun julia --color=yes --check-bounds=yes --project=.buildkite test/Spaces/unit_spaces.jl"
           - "srun julia --color=yes --check-bounds=yes --project=.buildkite test/Spaces/opt_spaces.jl"
-          - "julia --color=yes --check-bounds=yes --project=.buildkite test/Spaces/high_resolution_space.jl"
+          - "julia --color=yes --check-bounds=yes --project=.buildkite test/Spaces/unit_high_resolution_space.jl"
         env:
           CLIMACOMMS_DEVICE: "CUDA"
         agents:

test/Spaces/high_resolution_space.jl

@@ -0,0 +1,37 @@
+#=
+julia --project=.buildkite
+using Revise; include("test/Spaces/unit_high_resolution_space.jl")
+=#
+using Test
+import ClimaComms
+ClimaComms.@import_required_backends
+using ClimaCore.CommonSpaces
+using ClimaCore: Spaces
+
+# TODO: we should probably move these to datalayouts and just test high resolution kernels.
+@testset "Construct high resolution space" begin
+    space = ExtrudedCubedSphereSpace(
+        Float32;
+        radius = 1.0,
+        h_elem = 105,
+        z_elem = 10,
+        z_min = 1.0,
+        z_max = 2.0,
+        n_quad_points = 4,
+        staggering = CellCenter(),
+    )
+    @test space isa Spaces.CenterExtrudedFiniteDifferenceSpace
+
+    space =
+        CubedSphereSpace(Float32; radius = 1.0, n_quad_points = 4, h_elem = 105)
+    @test space isa Spaces.SpectralElementSpace2D
+
+    space = ColumnSpace(
+        Float32;
+        z_elem = 500,
+        z_min = 0,
+        z_max = 10,
+        staggering = CellCenter(),
+    )
+    @test space isa Spaces.FiniteDifferenceSpace
+end