add libcuml Python builds

rapidsai · Jan 22, 2025 · 9e1e0fe · 9e1e0fe
1 parent 5fb9720
commit 9e1e0fe
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/features/src/rapids-build-utils/devcontainer-feature.json b/features/src/rapids-build-utils/devcontainer-feature.json
@@ -1,7 +1,7 @@
 {
   "name": "NVIDIA RAPIDS devcontainer build utilities",
   "id": "rapids-build-utils",
-  "version": "25.2.6",
+  "version": "25.2.8",
   "description": "A feature to install the RAPIDS devcontainer build utilities",
   "containerEnv": {
     "BASH_ENV": "/etc/bash.bash_env"

diff --git a/features/src/rapids-build-utils/opt/rapids-build-utils/manifest.yaml b/features/src/rapids-build-utils/opt/rapids-build-utils/manifest.yaml
@@ -147,18 +147,23 @@ repos:
 
 - name: cuml
   path: cuml
-  git: {<<: *git_defaults, repo: cuml}
+  # TODO(jameslamb): revert this before merging
+  git: {<<: *git_defaults, repo: cuml, upstream: jameslamb, tag: libcuml-wheels}
   cpp:
     - name: cuml
       sub_dir: cpp
       depends: [rmm, raft, cumlprims_mg, cuvs]
       parallelism:
         max_device_obj_memory_usage: 3Gi
   python:
+    - name: libcuml
+      sub_dir: python/libcuml
+      depends: [cuml]
+      args: {install: *rapids_build_backend_args}
     - name: cuml
       sub_dir: python/cuml
       depends: [cuml]
-      args: {cmake: -DFIND_CUML_CPP=ON, install: *rapids_build_backend_args}
+      args: {install: *rapids_build_backend_args}
 
 - name: cugraph
   path: cugraph