initial work on adding wheel build for libcudf

rapidsai · Apr 10, 2024 · 87c60cd · 87c60cd
1 parent 7c3ad82
commit 87c60cd
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 0 deletions.
diff --git a/ci/build_wheel_cudf.sh b/ci/build_wheel_cudf.sh
@@ -14,4 +14,5 @@ if [[ ! -d "/tmp/gha-tools" ]]; then
     export PATH="/tmp/gha-tools/tools:${PATH}"
 fi
 
+RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen ${RAPIDS_CUDA_VERSION})"
 RAPIDS_PY_WHEEL_NAME="cudf_${RAPIDS_PY_CUDA_SUFFIX}" rapids-upload-wheels-to-s3 python ${package_dir}/final_dist
diff --git a/ci/build_wheel_libcudf.sh b/ci/build_wheel_libcudf.sh
@@ -12,4 +12,5 @@ if [[ ! -d "/tmp/gha-tools" ]]; then
     export PATH="/tmp/gha-tools/tools:${PATH}"
 fi
 
+RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen ${RAPIDS_CUDA_VERSION})"
 RAPIDS_PY_WHEEL_NAME="${package_name}_${RAPIDS_PY_CUDA_SUFFIX}" rapids-upload-wheels-to-s3 cpp dist
diff --git a/dependencies.yaml b/dependencies.yaml
@@ -82,8 +82,10 @@ files:
     extras:
       table: build-system
     includes:
+      - build_base
       - build_cpp
       - build_python_common
+      - build_python_cudf
   # This is the wrapper that gets used in Python, not the shared library wheel
   py_build_cudf:
     output: pyproject
@@ -321,9 +323,11 @@ dependencies:
           - matrix: {cuda: "12.*"}
             packages: &build_python_packages_cu12
               - &rmm_cu12 rmm-cu12==24.6.*
+              - &libcudf_cu12 libcudf-cu12==24.6.*
           - matrix: {cuda: "11.*"}
             packages: &build_python_packages_cu11
               - &rmm_cu11 rmm-cu11==24.6.*
+              - &libcudf_cu11 libcudf-cu11==24.6.*
           - {matrix: null, packages: null }
       - output_types: pyproject
         matrices:

diff --git a/python/libcudf/pyproject.toml b/python/libcudf/pyproject.toml
@@ -19,7 +19,9 @@ requires = [
     "cython>=3.0.3",
     "ninja",
     "numpy==1.23.*",
+    "protoc-wheel",
     "pyarrow==14.0.2.*",
+    "rmm==24.6.*",
     "scikit-build-core[pyproject]>=0.7.0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.