Skip to content

Commit 4ea6c1f

Browse files
committed
update all to gpu backend
1 parent 6da8cbe commit 4ea6c1f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

swm_python/lap_cartesian_vs_next.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -41,8 +41,8 @@
4141
import gt4py.next as gtx
4242

4343
#allocator = gtx.itir_python # should match the executor
44-
allocator = gtx.gtfn_cpu
45-
# allocator = gtx.gtfn_gpu
44+
#allocator = gtx.gtfn_cpu
45+
allocator = gtx.gtfn_gpu
4646

4747
# Note: for gt4py.next, names don't matter, for gt4py.cartesian they have to be "I", "J", "K"
4848
I = gtx.Dimension("I")

0 commit comments

Comments
 (0)