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

Support drop_last Argument in cuGraph-PyG Loader #3995

Merged
merged 48 commits into from
Nov 20, 2023
Merged
Changes from 1 commit
Commits
Show all changes
48 commits
Select commit Hold shift + click to select a range
e2b4764
fix large pandas dataframe
alexbarghi-nv Nov 1, 2023
f9305e5
c
alexbarghi-nv Nov 6, 2023
f1ef95f
fix loader bugs
alexbarghi-nv Nov 6, 2023
d5da756
test updates
alexbarghi-nv Nov 6, 2023
77c4525
style
alexbarghi-nv Nov 6, 2023
1a6124f
update dependencies.yaml
alexbarghi-nv Nov 7, 2023
c6a9724
Merge branch 'fix-loader-bugs' into fix-large-pandas-dataframe
alexbarghi-nv Nov 7, 2023
e207f98
Merge branch 'branch-23.12' into fix-loader-bugs
alexbarghi-nv Nov 7, 2023
20e4819
Merge branch 'branch-23.12' into fix-large-pandas-dataframe
alexbarghi-nv Nov 7, 2023
cf93084
style
alexbarghi-nv Nov 7, 2023
d7eb7d7
Merge branch 'fix-large-pandas-dataframe' of https://github.com/alexb…
alexbarghi-nv Nov 7, 2023
f619931
update test
alexbarghi-nv Nov 7, 2023
1f3cecf
drop last
alexbarghi-nv Nov 8, 2023
0f4693c
attempt to fix package resolution issue
alexbarghi-nv Nov 8, 2023
65f50a3
Merge branch 'fix-loader-bugs' of https://github.com/alexbarghi-nv/cu…
alexbarghi-nv Nov 8, 2023
9b7901c
Merge branch 'branch-23.12' into fix-loader-bugs
alexbarghi-nv Nov 8, 2023
9713cee
Add ucx to cugraph recipe
Nov 9, 2023
bc14fd7
Add ucx to libcugraph recipe
Nov 9, 2023
2064c37
Update conda/recipes/libcugraph/meta.yaml
naimnv Nov 9, 2023
b23a9d5
Merge branch 'fix_build' of https://github.com/naimnv/cugraph-forked …
alexbarghi-nv Nov 10, 2023
9384bb6
Merge branch 'fix-loader-bugs' of https://github.com/alexbarghi-nv/cu…
alexbarghi-nv Nov 10, 2023
11ff5bf
fix typing
alexbarghi-nv Nov 10, 2023
7b9396c
replace manual exchange with flip
alexbarghi-nv Nov 10, 2023
ede8399
Merge branch 'fix-loader-bugs' of https://github.com/alexbarghi-nv/cu…
alexbarghi-nv Nov 10, 2023
cf0f7f7
loader
alexbarghi-nv Nov 10, 2023
3920aac
complete merge
alexbarghi-nv Nov 10, 2023
0b05029
x
alexbarghi-nv Nov 10, 2023
801a302
node loader
alexbarghi-nv Nov 11, 2023
d1b5f22
various changes to get ci working
alexbarghi-nv Nov 11, 2023
e941e10
generator
alexbarghi-nv Nov 11, 2023
3f7ec7f
fix pylibcugraphops version, simplify test_python.sh
alexbarghi-nv Nov 12, 2023
46c4460
remove unwanted files
alexbarghi-nv Nov 12, 2023
a080f5a
Merge branch 'fix-loader-bugs' into fix-large-pandas-dataframe
alexbarghi-nv Nov 12, 2023
d229de4
remove strict csr/csr check
alexbarghi-nv Nov 12, 2023
55516bd
remove strict format check
alexbarghi-nv Nov 12, 2023
858b9d6
merge
alexbarghi-nv Nov 12, 2023
a6c43ae
Merge branch 'fix-loader-bugs' into drop-last
alexbarghi-nv Nov 12, 2023
9218d9c
add test for drop_last
alexbarghi-nv Nov 12, 2023
d1334ac
test correction
alexbarghi-nv Nov 14, 2023
2706d87
style
alexbarghi-nv Nov 14, 2023
d8e5eea
resolve merge conflict
alexbarghi-nv Nov 16, 2023
b9a9af3
revert meta changes
alexbarghi-nv Nov 16, 2023
4c9b73b
resolve merge conflict
alexbarghi-nv Nov 16, 2023
3350037
Merge branch 'fix-large-pandas-dataframe' into drop-last
alexbarghi-nv Nov 16, 2023
46d2960
fixes for small batch size
alexbarghi-nv Nov 17, 2023
e49ad92
resolve merge conflict
alexbarghi-nv Nov 17, 2023
9b214b2
Merge branch 'branch-23.12' into drop-last
alexbarghi-nv Nov 20, 2023
792c181
remove whitespace
alexbarghi-nv Nov 20, 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
Prev Previous commit
Next Next commit
attempt to fix package resolution issue
alexbarghi-nv committed Nov 8, 2023
commit 0f4693cd793131ea101930b5ec2063815e191828
14 changes: 10 additions & 4 deletions ci/test_python.sh
Original file line number Diff line number Diff line change
@@ -200,15 +200,21 @@ if [[ "${RAPIDS_CUDA_VERSION}" == "11.8.0" ]]; then
# Install pytorch
rapids-mamba-retry install \
--force-reinstall \
--channel pyg \
--channel pytorch \
--channel nvidia \
'pyg=2.3' \
'pytorch=2.0.0' \
--channel conda-forge \
'pytorch=2.1.0' \
'pytorch-cuda=11.8'

# Install pyg dependencies (which requires pip)
pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.0.0+cu118.html
pip install \
torch_geometric==2.4.0
pyg_lib \
torch_scatter \
torch_sparse \
torch_cluster \
torch_spline_conv \
-f https://data.pyg.org/whl/torch-2.1.0+cu118.html

rapids-mamba-retry install \
--channel "${CPP_CHANNEL}" \