Update test_reactant.jl

CliMA · Feb 4, 2025 · 1d04652 · 1d04652
1 parent 3f067cf
commit 1d04652
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/test/test_reactant.jl b/test/test_reactant.jl
@@ -1,4 +1,9 @@
 using Reactant
+if haskey(ENV, "GPU_TEST")
+    Reactant.set_default_backend("gpu")
+else
+    Reactant.set_default_backend("cpu")
+end
 using Test
 using Oceananigans
 using Oceananigans.Architectures