From d3c7a266b07f2b8177befe6084e50282784c6e4e Mon Sep 17 00:00:00 2001
From: Kyle Edwards <kyedwards@nvidia.com>
Date: Wed, 21 Aug 2024 14:34:23 -0400
Subject: [PATCH 1/2] Add use_cuda_wheels matrix entry

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.
---
 dependencies.yaml          | 13 ++++++++++++-
 python/cuml/pyproject.toml |  2 +-
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/dependencies.yaml b/dependencies.yaml
index e22aec7034..e1506abf3c 100644
--- a/dependencies.yaml
+++ b/dependencies.yaml
@@ -88,7 +88,7 @@ files:
     extras:
       table: project
     includes:
-      - cuda
+      - cuda_wheels
       - py_run
   py_test:
     output: pyproject
@@ -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
@@ -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
diff --git a/python/cuml/pyproject.toml b/python/cuml/pyproject.toml
index 0b5c650703..ad8d16e076 100644
--- a/python/cuml/pyproject.toml
+++ b/python/cuml/pyproject.toml
@@ -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",

From d8bc901823e334f5d72e8ddc025bda3a25e15c6a Mon Sep 17 00:00:00 2001
From: Kyle Edwards <kyedwards@nvidia.com>
Date: Wed, 21 Aug 2024 17:06:39 -0400
Subject: [PATCH 2/2] Add note about CUDA 11's lack of wheels

---
 dependencies.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/dependencies.yaml b/dependencies.yaml
index e1506abf3c..8c2508ce6b 100644
--- a/dependencies.yaml
+++ b/dependencies.yaml
@@ -420,6 +420,7 @@ dependencies:
               - nvidia-curand-cu12
               - nvidia-cusparse-cu12
               - nvidia-cusolver-cu12
+          # CUDA 11 does not provide wheels, so use the system libraries instead
           - matrix:
               cuda: "11.*"
               use_cuda_wheels: "true"