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.

Sign up for GitHub

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

Jump to bottom

[BUG]: cugraph-dgl example crashed #3847

Closed
2 tasks done
Tracked by #3806
chang-l opened this issue Sep 6, 2023 · 1 comment · Fixed by #3877
Closed
2 tasks done
Tracked by #3806

[BUG]: cugraph-dgl example crashed #3847

chang-l opened this issue Sep 6, 2023 · 1 comment · Fixed by #3877
Assignees
@alexbarghi-nv
Labels
bug Something isn't working
Milestone
23.10

Comments

@chang-l
Copy link

chang-l commented Sep 6, 2023

Version

22.10

Which installation method(s) does this occur on?

Source

Describe the bug.

According to README, the following cmd failed:
python3 node_classification.py --mode=gpu_cugraph_dgl

Minimum reproducible example

No response

Relevant log output

Traceback (most recent call last):
  File "/opt/dgl/qa/L1_cuGraph_GraphSAGE_1GPU/node.py", line 230, in <module>
    set_allocators()
  File "/opt/dgl/qa/L1_cuGraph_GraphSAGE_1GPU/node.py", line 48, in set_allocators
    torch.cuda.memory.change_current_allocator(rmm.rmm_torch_allocator)
AttributeError: module 'rmm' has no attribute 'rmm_torch_allocator'

Environment details

No response

Other/Misc.

No response

Code of Conduct

  • I agree to follow cuGraph's Code of Conduct
  • I have searched the open bugs and have found no duplicates for this bug report
@chang-l chang-l added ? - Needs Triage Need team to review and classify bug Something isn't working labels Sep 6, 2023
@chang-l
Copy link
Author

chang-l commented Sep 6, 2023

Looks like we should import rmm_torch_allocator as:

from rmm.allocators.torch import rmm_torch_allocator

@alexbarghi-nv alexbarghi-nv self-assigned this Sep 6, 2023
@alexbarghi-nv alexbarghi-nv removed the ? - Needs Triage Need team to review and classify label Sep 6, 2023
@alexbarghi-nv alexbarghi-nv added this to the 23.10 milestone Sep 6, 2023
This was referenced Sep 20, 2023
Open
Merged
rapids-bot bot pushed a commit that referenced this issue Sep 26, 2023
@alexbarghi-nv
Update Allocator Selection in cuGraph-DGL Example (#3877)
5c34d3d 
Closes #3847

Authors:
  - Alex Barghi (https://github.com/alexbarghi-nv)

Approvers:
  - Vibhu Jawa (https://github.com/VibhuJawa)
  - Brad Rees (https://github.com/BradReesWork)

URL: #3877
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@alexbarghi-nv alexbarghi-nv

Labels
bug Something isn't working
Projects
None yet
Milestone
23.10
Development

Successfully merging a pull request may close this issue.

Update Allocator Selection in cuGraph-DGL Example alexbarghi-nv/cugraph
2 participants
@chang-l @alexbarghi-nv