diff --git a/test/test_benchmark_performance.py b/test/test_benchmark_performance.py
index 830c6b9..049aa0b 100644
--- a/test/test_benchmark_performance.py
+++ b/test/test_benchmark_performance.py
@@ -11,10 +11,11 @@
     tile_put_sharded,
 )
 
+from pyscf_ipu.experimental.device import has_ipu, ipu_func
 from pyscf_ipu.nanoDFT.nanoDFT import build_mol, nanoDFT, nanoDFT_options
 
 
-@pytest.mark.skip(reason="No IPU in CI.")
+@pytest.mark.skipif(not has_ipu(), reason="Skipping ipu test!")
 @pytest.mark.ipu
 def test_basic_demonstration():
     dummy = np.random.rand(2,3).astype(np.float32)
@@ -40,7 +41,7 @@ def jitted_inner_test(dummy, dummier):
 
     assert True
 
-@pytest.mark.skip(reason="No IPU in CI.")
+@pytest.mark.skipif(not has_ipu(), reason="Skipping ipu test!")
 @pytest.mark.ipu
 @pytest.mark.parametrize("molecule", ["methane", "benzene"])
 def test_dense_eri(molecule):