add use_cuda_wheels

rapidsai · Jan 14, 2025 · 002c0f3 · 002c0f3
1 parent 53c8b35
commit 002c0f3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/dependencies.yaml b/dependencies.yaml
@@ -123,6 +123,7 @@ files:
     extras:
       table: project
     includes:
+      - cuda_wheels
       - depends_on_cuda_python
       - depends_on_cudf
       - depends_on_cupy

diff --git a/python/cuml/pyproject.toml b/python/cuml/pyproject.toml
@@ -102,6 +102,11 @@ dependencies = [
     "libcuml==25.2.*,>=0.0.0a0",
     "numba>=0.57",
     "numpy>=1.23,<3.0a0",
+    "nvidia-cublas",
+    "nvidia-cufft",
+    "nvidia-curand",
+    "nvidia-cusolver",
+    "nvidia-cusparse",
     "packaging",
     "pylibraft==25.2.*,>=0.0.0a0",
     "raft-dask==25.2.*,>=0.0.0a0",
@@ -171,7 +176,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",