Skip to content

Commit

Permalink
Add use_cuda_wheels matrix entry
Browse files Browse the repository at this point in the history
We want to be able to control whether or not the wheel uses the
CUDA wheels. Add a use_cuda_wheels matrix entry to control this.
  • Loading branch information
KyleFromNVIDIA committed Aug 21, 2024
1 parent d09e713 commit d3c7a26
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 2 deletions.
13 changes: 12 additions & 1 deletion dependencies.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ files:
extras:
table: project
includes:
- cuda
- cuda_wheels
- py_run
py_test:
output: pyproject
Expand Down Expand Up @@ -407,10 +407,13 @@ dependencies:
- *libcusolver114
- *libcusparse_dev114
- *libcusparse114
cuda_wheels:
specific:
- output_types: pyproject
matrices:
- matrix:
cuda: "12.*"
use_cuda_wheels: "true"
packages:
- nvidia-cublas-cu12
- nvidia-cufft-cu12
Expand All @@ -419,7 +422,15 @@ dependencies:
- nvidia-cusolver-cu12
- matrix:
cuda: "11.*"
use_cuda_wheels: "true"
packages:
# if use_cuda_wheels=false is provided, do not add dependencies on any CUDA wheels
# (e.g. for DLFW and pip devcontainers)
- matrix:
use_cuda_wheels: "false"
packages:
# if no matching matrix selectors passed, list the unsuffixed packages
# (just as a source of documentation, as this populates pyproject.toml in source control)
- matrix:
packages:
- nvidia-cublas
Expand Down
2 changes: 1 addition & 1 deletion python/cuml/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -159,7 +159,7 @@ versioneer\.py |
[tool.rapids-build-backend]
build-backend = "scikit_build_core.build"
dependencies-file = "../../dependencies.yaml"
matrix-entry = "cuda_suffixed=true"
matrix-entry = "cuda_suffixed=true;use_cuda_wheels=true"
requires = [
"cmake>=3.26.4,!=3.30.0",
"cuda-python",
Expand Down

0 comments on commit d3c7a26

Please sign in to comment.