Fix device

CliMA · Jun 22, 2023 · 005b898 · 005b898
1 parent b42bfe8
commit 005b898
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/Spaces/terrain_warp.jl b/test/Spaces/terrain_warp.jl
@@ -46,7 +46,8 @@ function generate_base_spaces(
     stretch = Meshes.Uniform();
     ndims = 3,
 )
-    comms_context = ClimaComms.SingletonCommsContext(ClimaComms.CPUDevice())
+    device = ClimaComms.CPUSingleThreaded()
+    comms_context = ClimaComms.SingletonCommsContext(device)
     FT = eltype(xlim)
     vertdomain = Domains.IntervalDomain(
         Geometry.ZPoint{FT}(zlim[1]),