change skipping so that it checks if there is IPU

graphcore-research · Oct 27, 2023 · a3d91ff · a3d91ff
1 parent 3441cb8
commit a3d91ff
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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):