hard-code skipping benchmark tests in CI

graphcore-research · Oct 27, 2023 · 16e8c36 · 16e8c36
1 parent 1233b13
commit 16e8c36
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/test/test_benchmark_performance.py b/test/test_benchmark_performance.py
@@ -4,11 +4,10 @@
 import pytest
 from tessellate_ipu import ipu_cycle_count, tile_map, tile_put_sharded
 
-from pyscf_ipu.experimental.device import has_ipu
 from pyscf_ipu.nanoDFT.nanoDFT import build_mol, nanoDFT, nanoDFT_options
 
 
-@pytest.mark.skipif(not has_ipu(), reason="Skipping ipu test!")
+@pytest.mark.skip(reason="Skipping IPU test in CI!")
 @pytest.mark.ipu
 def test_basic_demonstration():
     dummy = np.random.rand(2, 3).astype(np.float32)
@@ -35,7 +34,7 @@ def jitted_inner_test(dummy, dummier):
     assert True
 
 
-@pytest.mark.skipif(not has_ipu(), reason="Skipping ipu test!")
+@pytest.mark.skip(reason="Skipping IPU test in CI!")
 @pytest.mark.ipu
 @pytest.mark.parametrize("molecule", ["methane", "benzene"])
 def test_dense_eri(molecule):
@@ -62,7 +61,7 @@ def test_dense_eri(molecule):
     assert True
 
 
-@pytest.mark.skipif(not has_ipu(), reason="Skipping ipu test!")
+@pytest.mark.skip(reason="Skipping IPU test in CI!")
 @pytest.mark.ipu
 @pytest.mark.parametrize("molecule", ["methane", "benzene", "c20"])
 def test_sparse_eri(molecule):