rusty1s / pytorch_sparse

PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations
MIT License
1.01k stars 147 forks source link

Can't import torch_sparse after running setup.py #233

Closed berkekisin closed 2 years ago

berkekisin commented 2 years ago

Hi Matthias, I wanted to test "spmm_coo", so I ran "python setup.py install". It compiled without errors but as I tried to import torch_sparse I received the following error: Screenshot from 2022-05-24 14-28-37 I also tried to run setup.py and import torch_sparse without spmm_coo and I received the same error.

rusty1s commented 2 years ago

Can you show me the installation log of running

rm -rf build/
python setup.py clean
python setup.py develop
berkekisin commented 2 years ago

hello Matthias this worked thanks. I am ready to create PR. You said, "Provide general interface + documentation". Do you mean a "spmm_coo.py" file in directory torch_sparse and documentation there?

rusty1s commented 2 years ago

Hi @berkekisin. Yes, this is what I meant. We can set up a similar file structure as for our CSR-based spmm operation, e.g., spmm_coo.cpp and spmm_coo.py:

SauravMaheshkar commented 2 years ago

I assume the purpose of this issue is resolved ? CC: @rusty1s

rusty1s commented 2 years ago

Yes, related PR: https://github.com/rusty1s/pytorch_sparse/pull/237