Tests seem to no longer hang on gpu

lanl · Dec 15, 2023 · 77f9d08 · 77f9d08
1 parent 9e4f1b1
commit 77f9d08
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/scico/test/linop/xray/test_svmbir.py b/scico/test/linop/xray/test_svmbir.py
@@ -26,8 +26,6 @@
 BIG_INPUT = (32, 33, 50, 51, 125, 1.2)
 SMALL_INPUT = (4, 5, 7, 8, 16, 1.2)
 
-device = jax.devices()[0]
-
 
 def pytest_generate_tests(metafunc):
     param_ranges = {
@@ -154,7 +152,6 @@ def test_adjoint(
 
 
 @pytest.mark.slow
-@pytest.mark.skipif(device.platform != "cpu", reason="test hangs on gpu")
 def test_prox(
     is_3d,
     center_offset_small,
@@ -186,7 +183,6 @@ def test_prox(
 
 
 @pytest.mark.slow
-@pytest.mark.skipif(device.platform != "cpu", reason="test hangs on gpu")
 def test_prox_weights(
     is_3d,
     center_offset_small,