Merge branch 'master' of github.com:rusty1s/pytorch_sparse

torch_sparse/__init__.py

@@ -12,7 +12,7 @@ for library in [
     torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
         library, [osp.dirname(__file__)]).origin)
 
-if torch.version.cuda is not None:  # pragma: no cover
+if torch.cuda.is_available() and torch.version.cuda:  # pragma: no cover
     cuda_version = torch.ops.torch_sparse.cuda_version()
 
     if cuda_version == -1:

torch_sparse/reduce.py

@@ -35,7 +35,7 @@ def reduction(src: SparseTensor, dim: Optional[int] = None,
 
         if dim == 0 and value is not None:
             col = src.storage.col()
-            return scatter(value, col, dim=0, dim_size=src.size(0))
+            return scatter(value, col, 0, None, src.size(1), reduce)
         elif dim == 0 and value is None:
             if reduce == 'sum' or reduce == 'add':
                 return src.storage.colcount().to(src.dtype())

torch_sparse/tensor.py

@@ -66,7 +66,7 @@ class SparseTensor(object):
 
         value: Optional[torch.Tensor] = None
         if has_value:
-            value = mat._values()
+            value = mat.values()
 
         return SparseTensor(row=row, rowptr=None, col=col, value=value,
                             sparse_sizes=(mat.size(0), mat.size(1)),