Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Question] where does torch.ops.torch_sparse come from? #284

Closed
q-aaronzolnailucas opened this issue Oct 19, 2022 · 3 comments
Closed

[Question] where does torch.ops.torch_sparse come from? #284

q-aaronzolnailucas opened this issue Oct 19, 2022 · 3 comments
Labels

Comments

@q-aaronzolnailucas
Copy link

Trying to dive into how sparse-sparse matrix multiplication is working but I can't figure out where this import comes from, if I try importing it in a notebook with torch and torch_sparse imported I get an import error, but I see it's used in the source code of matmul.py. Where does this come from?

@rusty1s
Copy link
Owner

rusty1s commented Oct 29, 2022

The implementation of spspmm is here (CPU) and here (GPU). For GPU, we make use of cusparse routines, while we implement our own simple variant for CPU-processing.

@rusty1s
Copy link
Owner

rusty1s commented Oct 29, 2022

What's the issue you experience? Happy to help with installation :)

@github-actions
Copy link

This issue had no activity for 6 months. It will be closed in 2 weeks unless there is some new activity. Is this issue already resolved?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

No branches or pull requests

2 participants