AttributeError: module 'torch_quiver' has no attribute 'device_quiver_from_csr_array'

[LukeLIN-web]

I install torch-quiver 0.1.0 from https://github.com/quiver-team/torch-quiver/blob/main/docker/README.md when i run examples/pyg/reddit_quiver.py, it shows AttributeError: module 'torch_quiver' has no attribute 'device_quiver_from_csr_array'

[LukeLIN-web]

as well as in graph_sage_unsup_quiver.py

[LukeLIN-web]

I tried uninstall and install 0.1.1. It will hang at: https://github.com/quiver-team/torch-quiver/blob/0592669225954e46d9c57c0ca121c365af47b57f/examples/pyg/reddit_quiver.py#L29 as well as https://github.com/quiver-team/torch-quiver/blob/0592669225954e46d9c57c0ca121c365af47b57f/examples/pyg/ogbn_products_sage_quiver.py#L41

use pdb to find problem. it hang at srcs/python/quiver/pyg/sage_sampler.py self.quiver = qv.device_quiver_from_csr_array(self.csr_topo.indptr, self.csr_topo.indices, edge_id, device, self.mode != "UVA")

[ZenoTan]

Thanks for the ticket! https://github.com/quiver-team/torch-quiver/blob/main/srcs/cpp/src/quiver/cuda/quiver_sample.cu#L361-L461 The function calls this C++ function internally, and in your case it is hanging somewhere the middle.

[LukeLIN-web]

Thanks for the ticket! https://github.com/quiver-team/torch-quiver/blob/main/srcs/cpp/src/quiver/cuda/quiver_sample.cu#L361-L461 The function calls this C++ function internally, and in your case it is hanging somewhere the middle.

May you know how can i sort it out? Can I find any docker images that can run examples? I pull docker pull zenotan/torch_quiver:test. But it failed at AttributeError: Can't get attribute 'DataEdgeAttr' on <module 'torch_geometric.data.data' from '/opt/conda/lib/python3.7/site-packages/torch_geometric/data/data.py'>

[LukeLIN-web]

I figure it out, because I use a A100 GPU. compiled codes cannot run at A100. I change to V100 and run successfully.