Commit 6a992437 authored by rusty1s's avatar rusty1s
Browse files

skip test in case metis is not found

parent 3bda3da8
...@@ -4,7 +4,14 @@ from torch_sparse.tensor import SparseTensor ...@@ -4,7 +4,14 @@ from torch_sparse.tensor import SparseTensor
from .utils import devices from .utils import devices
try:
torch.ops.torch_sparse.partition
with_metis = True
except RuntimeError:
with_metis = False
@pytest.mark.skipif(not with_metis, reason='Not compiled with METIS support')
@pytest.mark.parametrize('device', devices) @pytest.mark.parametrize('device', devices)
def test_metis(device): def test_metis(device):
value1 = torch.randn(6 * 6, device=device).view(6, 6) value1 = torch.randn(6 * 6, device=device).view(6, 6)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment