Runtests on GPU

Author: Sbozzolo

.buildkite/pipeline.yml

@@ -69,6 +69,14 @@ steps:
         agents:
           slurm_gpus: 1
 
+      - label: "Unit: runtest cuda"
+        key: unit_runtest_cuda
+        command:
+          - "julia --project=test test/runtests.jl"
+        agents:
+          slurm_gpus: 1
+          slurm_mem: 20G
+
   - group: "Unit: RecursiveApply"
     steps:
 

test/Operators/spectralelement/opt.jl

@@ -13,6 +13,7 @@ import ClimaCore:
     Operators,
     Quadratures
 
+const context = ClimaComms.context()
 
 # We need to pull these broadcasted expressions out as
 # toplevel functions due to how broadcast expressions are
@@ -145,13 +146,8 @@ end
             quad = Quadratures.GLL{Nq}()
             mesh = Meshes.RectilinearMesh(domain, 3, 3)
 
-            topology = Topologies.Topology2D(
-                ClimaComms.SingletonCommsContext(
-                    ClimaComms.CPUSingleThreaded(),
-                ),
-                mesh,
-            )
-            space = Spaces.SpectralElementSpace2D(topology, quad)
+            topology = Topologies.Topology2D(mesh)
+            space = Spaces.SpectralElementSpace2D(context, topology, quad)
 
             coords = Fields.coordinate_field(space)
 
@@ -204,12 +200,7 @@ end
 
             horzdomain = Domains.RectangleDomain(xdomain, ydomain)
             horzmesh = Meshes.RectilinearMesh(horzdomain, xelem, yelem)
-            horztopology = Topologies.Topology2D(
-                ClimaComms.SingletonCommsContext(
-                    ClimaComms.CPUSingleThreaded(),
-                ),
-                horzmesh,
-            )
+            horztopology = Topologies.Topology2D(context, horzmesh)
 
             quad = Quadratures.GLL{npoly + 1}()
             horzspace = Spaces.SpectralElementSpace2D(horztopology, quad)