coalesce call

cuda/matmul_cuda.cu

@@ -28,6 +28,9 @@ static void init_cusparse() {
 }
 
 std::tuple<at::Tensor, at::Tensor> spspmm_cuda(at::Tensor A, at::Tensor B) {
+  A = A.coalesce();
+  B = B.coalesce();
+
   init_cusparse();
 
   auto m = A.size(0);

torch_sparse/sparse.py

-import torch
-
-
-def SparseTensor(index, value, size):
-    t = torch.cuda if value.is_cuda else torch
-    SparseTensor = getattr(t.sparse, value.type().split('.')[-1])
-    return SparseTensor(index, value, size)