We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3bda3da commit 6a99243Copy full SHA for 6a99243
test/test_metis.py
@@ -4,7 +4,14 @@
4
5
from .utils import devices
6
7
+try:
8
+ torch.ops.torch_sparse.partition
9
+ with_metis = True
10
+except RuntimeError:
11
+ with_metis = False
12
13
+
14
+@pytest.mark.skipif(not with_metis, reason='Not compiled with METIS support')
15
@pytest.mark.parametrize('device', devices)
16
def test_metis(device):
17
value1 = torch.randn(6 * 6, device=device).view(6, 6)
0 commit comments