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

Adds benchmarks for nx-cugraph #3854

Merged
Merged
Changes from 1 commit
Commits
Show all changes
50 commits
Select commit Hold shift + click to select a range
a63d350
Adds initial benchmarks for cugraph-nx, still WIP.
rlratzel Aug 24, 2023
87045be
Adds benchmarks specifically for larger datasets that use the k param…
rlratzel Aug 26, 2023
c08d603
Update the docstrings of the similarity algorithms (#3817)
jnke2016 Aug 25, 2023
3448b96
Use `copy-pr-bot` (#3827)
ajschmidt8 Aug 29, 2023
eb7de38
Disable mg tests (#3833)
naimnv Aug 30, 2023
0b79ea3
Fix OD shortest distance matrix computation test failures. (#3813)
seunghwak Aug 30, 2023
3c8a8c6
Remove legacy betweenness centrality (#3829)
jnke2016 Aug 30, 2023
b496254
Update README.md (#3826)
lmeyerov Aug 30, 2023
f270817
Add `louvain_communities` to cugraph-nx (#3803)
eriknw Aug 31, 2023
d909d8d
[BUG] Fix Batch Renumbering of Empty Batches (#3823)
alexbarghi-nv Aug 31, 2023
c0df6e2
Simplify wheel build scripts and allow alphas of RAPIDS dependencies …
vyasr Aug 31, 2023
262e281
Remove Deprecated Sampling Options (#3816)
alexbarghi-nv Sep 1, 2023
ccc8653
Use new `raft::compiled_static` targets (#3842)
divyegala Sep 6, 2023
7c5f38b
[IMP] Add ability to get batch size from the loader in cuGraph-PyG (#…
alexbarghi-nv Sep 6, 2023
80b7ae0
Rename `cugraph-nx` to `nx-cugraph` (#3840)
eriknw Sep 6, 2023
a24341d
Migrate upstream models to `cugraph-pyg` (#3763)
tingyu66 Sep 6, 2023
5574eb6
Expose threshold in louvain (#3792)
ChuckHastings Sep 6, 2023
a2972b5
Remove the assumption made on the client data's keys (#3835)
jnke2016 Sep 7, 2023
4b18ddf
Adding metadata getter methods to datasets API (#3821)
nv-rliu Sep 8, 2023
6282965
Uses `conda mambabuild` rather than `mamba mambabuild` (#3853)
rlratzel Sep 8, 2023
5babc71
Merge remote-tracking branch 'upstream/branch-23.10' into branch-23.1…
rlratzel Sep 8, 2023
c3194c4
Renames dir to nx-cugraph for consistency with new package name.
rlratzel Sep 8, 2023
96dc4f2
Merge remote-tracking branch 'upstream/branch-23.10' into branch-23.1…
rlratzel Sep 26, 2023
5091a8b
Merge remote-tracking branch 'upstream/branch-23.10' into branch-23.1…
rlratzel Sep 27, 2023
9248af0
Adds benchmark for louvain using small graphs, adds support for Netwo…
rlratzel Sep 27, 2023
15b28ee
Adds benchmark for louvain_communities using medium size graphs.
rlratzel Sep 27, 2023
c458918
Removed unused imports, adds comment describing fixture args (ids, et…
rlratzel Sep 28, 2023
eb703ac
Merge remote-tracking branch 'upstream/branch-23.10' into branch-23.1…
rlratzel Sep 28, 2023
ca5e255
Merge remote-tracking branch 'upstream/branch-23.12' into branch-23.1…
rlratzel Sep 28, 2023
1232897
Removes FIXME, minor code cleanup.
rlratzel Sep 28, 2023
880ff6b
Initial black run on benchmark files.
rlratzel Sep 29, 2023
ade2937
Merge branch 'branch-23.10-cugraph_nx_benchmarks' of github.com:rlrat…
trxcllnt Sep 29, 2023
8295fd3
bump versions 23.10 -> 23.12
trxcllnt Sep 29, 2023
c8c8dc0
leave one thread free
trxcllnt Sep 29, 2023
f845799
Merge branch 'branch-23.12' of github.com:rapidsai/cugraph into rlrat…
trxcllnt Sep 29, 2023
7d6b350
fix dependencies.yaml merge from 23.10 -> 23.12
trxcllnt Sep 29, 2023
3fdb9ba
update ucx-py, dask, and distributed versions
trxcllnt Sep 29, 2023
c827b81
separate CUDA suffixes for pylibcugraphops
trxcllnt Sep 29, 2023
2ec390e
Merge branch 'branch-23.12' of github.com:rapidsai/cugraph into rlrat…
trxcllnt Sep 29, 2023
c00c3f1
WholeGraph Feature Store for cuGraph-PyG and cuGraph-DGL (#3874)
alexbarghi-nv Sep 30, 2023
dc87455
increase timeout
jnke2016 Oct 3, 2023
a863835
Integrate renumbering and compression to `cugraph-dgl` to accelerate …
tingyu66 Oct 3, 2023
a564957
add env var to the wheel run
jnke2016 Oct 3, 2023
3f1547f
Merge remote-tracking branch 'upstream/branch-23.10' into branch-23.1…
jnke2016 Oct 3, 2023
74c63fb
increase timeouts
jnke2016 Oct 4, 2023
8f00bbd
Adds k as a param for BC bench.
rlratzel Oct 6, 2023
9955b7c
Merge branch 'branch-23.10-cugraph_nx_benchmarks' of https://github.c…
rlratzel Oct 6, 2023
bf74a73
Merge remote-tracking branch 'upstream/branch-23.12' into branch-23.1…
rlratzel Oct 6, 2023
cbdbd8a
Merge remote-tracking branch 'jnke2016/branch-23.10_increase-timeout'…
rlratzel Oct 6, 2023
80b9cf2
Updates pylibwholegraph version to 23.12.
rlratzel Oct 6, 2023
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
49 changes: 19 additions & 30 deletions dependencies.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -341,7 +341,7 @@ dependencies:
- sphinx-markdown-tables
- sphinx<6
- sphinxcontrib-websupport
- &pylibcugraphops pylibcugraphops==23.12.*
- pylibcugraphops==23.12.*
rlratzel marked this conversation as resolved.
Show resolved Hide resolved
py_version:
specific:
- output_types: [conda]
Expand All @@ -368,27 +368,16 @@ dependencies:
- output_types: [conda, pyproject, requirements]
packages:
- cython>=3.0.0
- &pylibraft pylibraft==23.12.*
- &rmm rmm==23.12.*
- scikit-build>=0.13.1
python_build_cugraph:
common:
- output_types: [conda, pyproject]
packages:
- &pylibcugraph pylibcugraph==23.12.*
python_run_cugraph:
common:
- output_types: [conda, pyproject]
packages:
- &cudf cudf==23.12.*
- &dask dask>=2023.7.1
- &distributed distributed>=2023.7.1
- &dask dask==2023.9.2
- &distributed distributed==2023.9.2
- &dask_cuda dask-cuda==23.12.*
- &dask_cudf dask-cudf==23.12.*
- &numba numba>=0.57
- raft-dask==23.12.*
- *rmm
- &ucx_py ucx-py==0.35.*
- &ucx_py ucx-py==0.34.*
- output_types: conda
packages:
- aiohttp
Expand Down Expand Up @@ -462,7 +451,7 @@ dependencies:
- sphinxcontrib-websupport
- sphinx-markdown-tables
- sphinx-copybutton
- *pylibcugraphops
- pylibcugraphops==23.12.*
test_cpp:
common:
- output_types: conda
Expand Down Expand Up @@ -509,7 +498,7 @@ dependencies:
- output_types: [conda]
packages:
- cugraph==23.12.*
- *pylibcugraphops
- pylibcugraphops==23.12.*
- pytorch>=2.0
- pytorch-cuda==11.8
- dgl>=1.1.0.cu*
Expand All @@ -518,7 +507,7 @@ dependencies:
- output_types: [conda]
packages:
- cugraph==23.12.*
- *pylibcugraphops
- pylibcugraphops==23.12.*
- pytorch==2.0
- pytorch-cuda==11.8
- pyg=2.3.1=*torch_2.0.0*cu118*
Expand All @@ -527,7 +516,7 @@ dependencies:
common:
- output_types: conda
packages:
- *rmm
- &rmm_conda rmm==23.12.*
- output_types: requirements
packages:
# pip recognizes the index as a global option for the requirements.txt file
Expand All @@ -546,13 +535,13 @@ dependencies:
- {matrix: {cuda: "11.5"}, packages: *rmm_packages_pip_cu11}
- {matrix: {cuda: "11.4"}, packages: *rmm_packages_pip_cu11}
- {matrix: {cuda: "11.2"}, packages: *rmm_packages_pip_cu11}
- {matrix: null, packages: [*rmm]}
- {matrix: null, packages: [*rmm_conda]}

depends_on_cudf:
common:
- output_types: conda
packages:
- *cudf
- &cudf_conda cudf==23.12.*
- output_types: requirements
packages:
# pip recognizes the index as a global option for the requirements.txt file
Expand All @@ -571,13 +560,13 @@ dependencies:
- {matrix: {cuda: "11.5"}, packages: *cudf_packages_pip_cu11}
- {matrix: {cuda: "11.4"}, packages: *cudf_packages_pip_cu11}
- {matrix: {cuda: "11.2"}, packages: *cudf_packages_pip_cu11}
- {matrix: null, packages: [*cudf]}
- {matrix: null, packages: [*cudf_conda]}

depends_on_dask_cudf:
common:
- output_types: conda
packages:
- *dask_cudf
- &dask_cudf_conda dask-cudf==23.12.*
- output_types: requirements
packages:
# pip recognizes the index as a global option for the requirements.txt file
Expand All @@ -596,13 +585,13 @@ dependencies:
- {matrix: {cuda: "11.5"}, packages: *dask_cudf_packages_pip_cu11}
- {matrix: {cuda: "11.4"}, packages: *dask_cudf_packages_pip_cu11}
- {matrix: {cuda: "11.2"}, packages: *dask_cudf_packages_pip_cu11}
- {matrix: null, packages: [*dask_cudf]}
- {matrix: null, packages: [*dask_cudf_conda]}

depends_on_pylibraft:
common:
- output_types: conda
packages:
- *pylibraft
- &pylibraft_conda pylibraft==23.12.*
- output_types: requirements
packages:
# pip recognizes the index as a global option for the requirements.txt file
Expand All @@ -621,13 +610,13 @@ dependencies:
- {matrix: {cuda: "11.5"}, packages: *pylibraft_packages_pip_cu11}
- {matrix: {cuda: "11.4"}, packages: *pylibraft_packages_pip_cu11}
- {matrix: {cuda: "11.2"}, packages: *pylibraft_packages_pip_cu11}
- {matrix: null, packages: [*pylibraft]}
- {matrix: null, packages: [*pylibraft_conda]}

depends_on_raft_dask:
common:
- output_types: conda
packages:
- &raft_dask raft-dask==23.12.*
- &raft_dask_conda raft-dask==23.12.*
- output_types: requirements
packages:
# pip recognizes the index as a global option for the requirements.txt file
Expand All @@ -646,13 +635,13 @@ dependencies:
- {matrix: {cuda: "11.5"}, packages: *raft_dask_packages_pip_cu11}
- {matrix: {cuda: "11.4"}, packages: *raft_dask_packages_pip_cu11}
- {matrix: {cuda: "11.2"}, packages: *raft_dask_packages_pip_cu11}
- {matrix: null, packages: [*raft_dask]}
- {matrix: null, packages: [*raft_dask_conda]}

depends_on_pylibcugraph:
common:
- output_types: conda
packages:
- *pylibcugraph
- &pylibcugraph_conda pylibcugraph==23.12.*
- output_types: requirements
packages:
# pip recognizes the index as a global option for the requirements.txt file
Expand All @@ -671,7 +660,7 @@ dependencies:
- {matrix: {cuda: "11.5"}, packages: *pylibcugraph_packages_pip_cu11}
- {matrix: {cuda: "11.4"}, packages: *pylibcugraph_packages_pip_cu11}
- {matrix: {cuda: "11.2"}, packages: *pylibcugraph_packages_pip_cu11}
- {matrix: null, packages: [*pylibcugraph]}
- {matrix: null, packages: [*pylibcugraph_conda]}

depends_on_cupy:
common:
Expand Down
Loading