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

Fix OOM Bug for Jaccard, Sorensen, and Overlap benchmarks #4524

Merged

Conversation

nv-rliu
Copy link
Contributor

@nv-rliu nv-rliu commented Jul 8, 2024

Closes #4510

This PR fixes a bug in benchmarks/cugraph/pytest-based/bench_algos.py for Jaccard and Overlap (and adds Sorensen to the list of algos to be run).

The reason for the error was that the benchmarks were trying to compute two_hop_neighbors on all nodes which would cause the GPU to run out of memory.

@nv-rliu nv-rliu added this to the 24.08 milestone Jul 8, 2024
@nv-rliu nv-rliu added bug Something isn't working non-breaking Non-breaking change labels Jul 8, 2024
@nv-rliu nv-rliu requested a review from jnke2016 July 8, 2024 15:32
@nv-rliu nv-rliu requested a review from rlratzel July 8, 2024 15:42
Copy link
Contributor

@jnke2016 jnke2016 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I left a suggestion regarding the vertex_pairs generation.

@nv-rliu nv-rliu requested a review from a team as a code owner July 9, 2024 01:59
@github-actions github-actions bot added the python label Jul 9, 2024
@nv-rliu nv-rliu requested review from jnke2016 and removed request for rlratzel July 9, 2024 02:00
Copy link
Contributor

@jnke2016 jnke2016 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good to me

Copy link
Member

@alexbarghi-nv alexbarghi-nv left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

👍

@alexbarghi-nv
Copy link
Member

/merge

@rapids-bot rapids-bot bot merged commit 4464504 into rapidsai:branch-24.08 Jul 9, 2024
131 checks passed
@nv-rliu nv-rliu deleted the 24.08-nightly-benchmark-fix branch July 9, 2024 18:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
benchmarks bug Something isn't working non-breaking Non-breaking change python
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Jaccard, Sorensen, and Overlap OOM in bench_algos.py
3 participants