@@ -13,6 +13,7 @@ import ClimaCore:
13
13
Operators,
14
14
Quadratures
15
15
16
+ const context = ClimaComms. context ()
16
17
17
18
# We need to pull these broadcasted expressions out as
18
19
# toplevel functions due to how broadcast expressions are
145
146
quad = Quadratures. GLL {Nq} ()
146
147
mesh = Meshes. RectilinearMesh (domain, 3 , 3 )
147
148
148
- topology = Topologies. Topology2D (
149
- ClimaComms. SingletonCommsContext (
150
- ClimaComms. CPUSingleThreaded (),
151
- ),
152
- mesh,
153
- )
154
- space = Spaces. SpectralElementSpace2D (topology, quad)
149
+ topology = Topologies. Topology2D (mesh)
150
+ space = Spaces. SpectralElementSpace2D (context, topology, quad)
155
151
156
152
coords = Fields. coordinate_field (space)
157
153
204
200
205
201
horzdomain = Domains. RectangleDomain (xdomain, ydomain)
206
202
horzmesh = Meshes. RectilinearMesh (horzdomain, xelem, yelem)
207
- horztopology = Topologies. Topology2D (
208
- ClimaComms. SingletonCommsContext (
209
- ClimaComms. CPUSingleThreaded (),
210
- ),
211
- horzmesh,
212
- )
203
+ horztopology = Topologies. Topology2D (context, horzmesh)
213
204
214
205
quad = Quadratures. GLL {npoly + 1} ()
215
206
horzspace = Spaces. SpectralElementSpace2D (horztopology, quad)
0 commit comments