enable some stablehlo tests

apache · Oct 25, 2024 · 9efd894 · 9efd894
1 parent 9772ed0
commit 9efd894
Showing 1 changed file with 0 additions and 24 deletions.
diff --git a/tests/python/relax/test_frontend_stablehlo.py b/tests/python/relax/test_frontend_stablehlo.py
@@ -196,10 +196,6 @@ def main(
 
 
 @tvm.testing.requires_gpu
-@pytest.mark.skip(
-    reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed."
-)
-# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
 def test_unary():
     import jax
 
@@ -233,10 +229,6 @@ def _round(x):
 
 
 @tvm.testing.requires_gpu
-@pytest.mark.skip(
-    reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed."
-)
-# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
 def test_binary():
     import jax
 
@@ -258,10 +250,6 @@ def fn(x, y):
 
 
 @tvm.testing.requires_gpu
-@pytest.mark.skip(
-    reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed."
-)
-# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
 def test_const():
     import jax
 
@@ -272,10 +260,6 @@ def fn(x):
 
 
 @tvm.testing.requires_gpu
-@pytest.mark.skip(
-    reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed."
-)
-# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
 def test_maximum():
     import jax
     import jax.numpy as jnp
@@ -287,10 +271,6 @@ def fn(x, y):
 
 
 @tvm.testing.requires_gpu
-@pytest.mark.skip(
-    reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed."
-)
-# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
 def test_minimum():
     import jax
     import jax.numpy as jnp
@@ -332,10 +312,6 @@ def fn(x):
 
 
 @tvm.testing.requires_gpu
-@pytest.mark.skip(
-    reason="jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed."
-)
-# TODO(mshr-h): may be fixed by upgrading jax to >=0.4.33
 def test_dot_general():
     import jax