upgrade to PyTorch 1.10

.github/workflows/testing.yml

@@ -11,7 +11,7 @@ jobs:
       matrix:
         os: [ubuntu-latest, windows-latest]
         python-version: [3.6]
-        torch-version: [1.8.0, 1.9.0]
+        torch-version: [1.9.0, 1.10.0]
 
     steps:
       - uses: actions/checkout@v2

setup.py

@@ -29,7 +29,7 @@ def get_extensions():
     for main, suffix in product(main_files, suffices):
         define_macros = []
         extra_compile_args = {'cxx': ['-O2']}
-        extra_link_args = ['-s']
+        extra_link_args = []
 
         if suffix == 'cuda':
             define_macros += [('WITH_CUDA', None)]

test/test_conv.py

@@ -58,7 +58,7 @@ def test_spline_conv_forward(test, dtype, device):
 
 
 @pytest.mark.parametrize('degree,device', product(degrees, devices))
-def test_spline_basis_backward(degree, device):
+def test_spline_conv_backward(degree, device):
     x = torch.rand((3, 2), dtype=torch.double, device=device)
     x.requires_grad_()
     edge_index = tensor([[0, 1, 1, 2], [1, 0, 2, 1]], torch.long, device)

torch_spline_conv/conv.py

@@ -69,10 +69,10 @@ def spline_conv(x: torch.Tensor, edge_index: torch.Tensor,
 
     # Weight root node separately (if wished).
     if root_weight is not None:
-        out = out + torch.matmul(x, root_weight)
+        out += x @ root_weight
 
     # Add bias (if wished).
     if bias is not None:
-        out = out + bias
+        out += bias
 
     return out