torch_sparse_solve

torch_sparse_solve copied to clipboard

Much slower than `torch.linalg.solve`

2

I wrote the following code to benchmark against `torch.linalg.solve` for large sparse matrices. Here, `A` is a 5000x5000 coalesced COO matrix with 10 nonzero elements in every row. `b` is...

uranium11010

Cannot install torch_sparse_solve

1

Dear flaport, I tried multiple times with different environmental settings in Colab, but I consistantly get the same error msg: Preparing metadata (setup.py) ... done error: subprocess-exited-with-error × python setup.py...

ysyysy115

Avoiding the for loop in solve

1

Hey Floris, I wonder if you'd consider removing the for-loop in solve by using a method similar to [this](https://pytorch-geometric.readthedocs.io/en/latest/notes/batching.html) i.e treating the whole system with a block diagonal A where...

mpariente

Supporting pytorch_sparse

1

Hi again, You might be aware of this project ([`pytorch_sparse`](https://github.com/rusty1s/pytorch_sparse)) which is really cool. It's be great to support their `SparseTensor` as an input to `solve`, or maybe adding another...

mpariente

fail to pass the `torch.autograd.gradcheck`

2

Thank you very much for providing such a good tool! My problem is that when the input `A` is a 'real' sparse matrix, not the sparse matrix converted from a...

zhf-0