try to build with metis

.travis.yml

@@ -60,6 +60,7 @@ jobs:
 install:
   - source script/cuda.sh
   - source script/conda.sh
+  - source script/metis.sh
   - conda create --yes -n test python="${PYTHON_VERSION}"
   - source activate test
   - conda install pytorch=${TORCH_VERSION} ${TOOLKIT} -c pytorch --yes

script/metis.sh

+#!/bin/bash
+
+METIS=metis-5.1.0
+
+wget -nv http://glaros.dtc.umn.edu/gkhome/fetch/sw/metis/${METIS}.tar.gz
+tar -xvzf ${METIS}.tar.gz
+cd ${METIS} || exit
+sed -i.bak -e 's/IDXTYPEWIDTH 32/IDXTYPEWIDTH 64/g' include/metis.h
+make config
+make
+make install

test/test_metis.py

+import pytest
+import torch
+from torch_sparse.tensor import SparseTensor
+
+from .utils import devices
+
+
+@pytest.mark.parametrize('device', devices)
+def test_metis(device):
+    mat = SparseTensor.from_dense(torch.randn((6, 6), device=device))
+    mat, partptr, perm = mat.partition_kway(num_parts=2)
+    assert partptr.tolist() == [0, 3, 6]
+    assert perm.tolist() == [0, 1, 2, 3, 4, 5]