Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
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
2 changes: 2 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,8 @@ wheels/
wheelhouse/
_skbuild/
cufile.log
*.tar.gz
*.whl

## Patching
*.diff
Expand Down
61 changes: 45 additions & 16 deletions dependencies.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -298,7 +298,7 @@ dependencies:
- output_types: [conda, pyproject, requirements]
packages:
- *numpy
- packaging
- &packaging packaging
- pandas
rapids_build_skbuild:
common:
Expand Down Expand Up @@ -331,19 +331,58 @@ dependencies:
- pytest-benchmark
- pytest-cov
- pytest-xdist
specific:
- output_types: [conda]
matrices:
- matrix:
no_pytorch: "true"
packages:
- matrix:
packages:
- torchdata
- pydantic
test_python_pylibwholegraph:
common:
- output_types: [conda, pyproject, requirements]
packages:
- *packaging
- pytest-forked
- scipy
depends_on_pytorch:
common:
- output_types: [conda]
packages:
- torchdata
- pydantic
depends_on_pytorch:
specific:
# conda: choose between GPU and CPU-only pytorch
- output_types: conda
matrices:
- matrix:
no_pytorch: "true"
packages:
- matrix:
require_gpu: "true"
packages:
- pytorch-gpu>=2.3
# Default to falling back to whatever 'pytorch' is pulled in via cugraph-pyg's dependencies.
- matrix:
packages:
# conda: optionally pin to older versions based on 'dependencies' key
- output_types: conda
matrices:
- matrix:
no_pytorch: "true"
packages:
- matrix:
cuda: "12.*"
dependencies: "oldest"
packages:
- pytorch==2.4.1
- matrix:
cuda: "13.*"
dependencies: "oldest"
packages:
- pytorch==2.9
# Default to falling back to whatever 'pytorch' is pulled in via cugraph-pyg's dependencies.
- matrix:
packages:
- output_types: [requirements]
matrices:
- matrix:
Expand Down Expand Up @@ -375,16 +414,6 @@ dependencies:
- matrix:
packages:
- *pytorch_pip
- output_types: [conda]
matrices:
# Prevent fallback to CPU-only pytorch when we want a CUDA variant.
- matrix:
require_gpu: "true"
packages:
- pytorch-gpu
# Default to falling back to whatever 'pytorch' is pulled in via cugraph-pyg's dependencies.
- matrix:
packages:
depends_on_nccl:
common:
- output_types: conda
Expand Down
1 change: 1 addition & 0 deletions python/pylibwholegraph/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,7 @@ dependencies = [

[project.optional-dependencies]
test = [
"packaging",
"pytest",
"pytest-benchmark",
"pytest-cov",
Expand Down