Merge pull request #44 from daxiongshu/branch-22.10

sync with upstream
daxiongshu · Sep 1, 2022 · f9015bb · f9015bb
2 parents c835151 + 78ba4e8
commit f9015bb
Show file tree

Hide file tree

Showing 222 changed files with 8,009 additions and 1,446 deletions.
diff --git a/.github/ops-bot.yaml b/.github/ops-bot.yaml
@@ -5,4 +5,5 @@ auto_merger: true
 branch_checker: true
 label_checker: true
 release_drafter: true
-external_contributors: false
+copy_prs: true
+rerun_tests: true
diff --git a/.gitignore b/.gitignore
@@ -30,6 +30,8 @@ log
 dask-worker-space/
 tmp/
 .hypothesis
+wheels/
+_skbuild/
 
 ## files pickled in notebook when ran during python docstring generation
 docs/source/*.model

diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,6 +1,63 @@
-# cuML 22.06.00 (Date TBD)
+# cuML 22.10.00 (Date TBD)
 
-Please see https://github.com/rapidsai/cuml/releases/tag/v22.06.00a for the latest changes to this development branch.
+Please see https://github.com/rapidsai/cuml/releases/tag/v22.10.00a for the latest changes to this development branch.
+
+# cuML 22.08.00 (Date TBD)
+
+Please see https://github.com/rapidsai/cuml/releases/tag/v22.08.00a for the latest changes to this development branch.
+
+# cuML 22.06.00 (7 Jun 2022)
+
+## 🐛 Bug Fixes
+
+- Fix sg benchmark build. ([#4766](https://github.com/rapidsai/cuml/pull/4766)) [@trivialfis](https://github.com/trivialfis)
+- Resolve KRR hypothesis test failure ([#4761](https://github.com/rapidsai/cuml/pull/4761)) [@RAMitchell](https://github.com/RAMitchell)
+- Fix `KBinsDiscretizer` `bin_edges_` ([#4735](https://github.com/rapidsai/cuml/pull/4735)) [@viclafargue](https://github.com/viclafargue)
+- FIX Accept small floats in RandomForest ([#4717](https://github.com/rapidsai/cuml/pull/4717)) [@thomasjpfan](https://github.com/thomasjpfan)
+- Remove import of `scalar_broadcast_to` from stemmer ([#4706](https://github.com/rapidsai/cuml/pull/4706)) [@viclafargue](https://github.com/viclafargue)
+- Replace 22.04.x with 22.06.x in yaml files ([#4692](https://github.com/rapidsai/cuml/pull/4692)) [@daxiongshu](https://github.com/daxiongshu)
+- Replace cudf.logical_not with ~ ([#4669](https://github.com/rapidsai/cuml/pull/4669)) [@canonizer](https://github.com/canonizer)
+
+## 📖 Documentation
+
+- Fix docs builds ([#4733](https://github.com/rapidsai/cuml/pull/4733)) [@ajschmidt8](https://github.com/ajschmidt8)
+- Change &quot;principals&quot; to &quot;principles&quot; ([#4695](https://github.com/rapidsai/cuml/pull/4695)) [@cakiki](https://github.com/cakiki)
+- Update pydoc and promote `ColumnTransformer` out of experimental ([#4509](https://github.com/rapidsai/cuml/pull/4509)) [@viclafargue](https://github.com/viclafargue)
+
+## 🚀 New Features
+
+- float64 support in FIL functions ([#4655](https://github.com/rapidsai/cuml/pull/4655)) [@canonizer](https://github.com/canonizer)
+- float64 support in FIL core ([#4646](https://github.com/rapidsai/cuml/pull/4646)) [@canonizer](https://github.com/canonizer)
+- Allow &quot;LabelEncoder&quot; to accept cupy and numpy arrays as input. ([#4620](https://github.com/rapidsai/cuml/pull/4620)) [@daxiongshu](https://github.com/daxiongshu)
+- MNMG Logistic Regression (dask-glm wrapper) ([#3512](https://github.com/rapidsai/cuml/pull/3512)) [@daxiongshu](https://github.com/daxiongshu)
+
+## 🛠️ Improvements
+
+- Pin `dask` &amp; `distributed` for release ([#4758](https://github.com/rapidsai/cuml/pull/4758)) [@galipremsagar](https://github.com/galipremsagar)
+- Simplicial set functions ([#4756](https://github.com/rapidsai/cuml/pull/4756)) [@viclafargue](https://github.com/viclafargue)
+- Upgrade Treelite to 2.4.0 ([#4752](https://github.com/rapidsai/cuml/pull/4752)) [@hcho3](https://github.com/hcho3)
+- Simplify recipes ([#4749](https://github.com/rapidsai/cuml/pull/4749)) [@Ethyling](https://github.com/Ethyling)
+- Inference for float64 random forests using FIL ([#4739](https://github.com/rapidsai/cuml/pull/4739)) [@canonizer](https://github.com/canonizer)
+- MNT Removes unused optim_batch_size from UMAP&#39;s docstring ([#4732](https://github.com/rapidsai/cuml/pull/4732)) [@thomasjpfan](https://github.com/thomasjpfan)
+- Require UCX 1.12.1+ ([#4720](https://github.com/rapidsai/cuml/pull/4720)) [@jakirkham](https://github.com/jakirkham)
+- Allow enabling raft NVTX markers when raft is installed ([#4718](https://github.com/rapidsai/cuml/pull/4718)) [@achirkin](https://github.com/achirkin)
+- Fix identifier collision ([#4716](https://github.com/rapidsai/cuml/pull/4716)) [@viclafargue](https://github.com/viclafargue)
+- Use raft::span in TreeExplainer ([#4714](https://github.com/rapidsai/cuml/pull/4714)) [@hcho3](https://github.com/hcho3)
+- Expose simplicial set functions ([#4711](https://github.com/rapidsai/cuml/pull/4711)) [@viclafargue](https://github.com/viclafargue)
+- Refactor `tests` in `cuml` ([#4703](https://github.com/rapidsai/cuml/pull/4703)) [@galipremsagar](https://github.com/galipremsagar)
+- Use conda to build python packages during GPU tests ([#4702](https://github.com/rapidsai/cuml/pull/4702)) [@Ethyling](https://github.com/Ethyling)
+- Update pinning to allow newer CMake versions. ([#4698](https://github.com/rapidsai/cuml/pull/4698)) [@vyasr](https://github.com/vyasr)
+- TreeExplainer extensions ([#4697](https://github.com/rapidsai/cuml/pull/4697)) [@RAMitchell](https://github.com/RAMitchell)
+- Add sample_weight for Ridge ([#4696](https://github.com/rapidsai/cuml/pull/4696)) [@lowener](https://github.com/lowener)
+- Unpin `dask` &amp; `distributed` for development ([#4693](https://github.com/rapidsai/cuml/pull/4693)) [@galipremsagar](https://github.com/galipremsagar)
+- float64 support in treelite-&gt;FIL import and Python layer ([#4690](https://github.com/rapidsai/cuml/pull/4690)) [@canonizer](https://github.com/canonizer)
+- Enable building static libs ([#4673](https://github.com/rapidsai/cuml/pull/4673)) [@trxcllnt](https://github.com/trxcllnt)
+- Treeshap hypothesis tests ([#4671](https://github.com/rapidsai/cuml/pull/4671)) [@RAMitchell](https://github.com/RAMitchell)
+- float64 support in multi-sum and child_index() ([#4648](https://github.com/rapidsai/cuml/pull/4648)) [@canonizer](https://github.com/canonizer)
+- Add libcuml-tests package ([#4635](https://github.com/rapidsai/cuml/pull/4635)) [@Ethyling](https://github.com/Ethyling)
+- Random ball cover algorithm for 3D data ([#4582](https://github.com/rapidsai/cuml/pull/4582)) [@cjnolet](https://github.com/cjnolet)
+- Use conda compilers ([#4577](https://github.com/rapidsai/cuml/pull/4577)) [@Ethyling](https://github.com/Ethyling)
+- Build packages using mambabuild ([#4542](https://github.com/rapidsai/cuml/pull/4542)) [@Ethyling](https://github.com/Ethyling)
 
 # cuML 22.04.00 (6 Apr 2022)
 

diff --git a/build.sh b/build.sh
@@ -229,7 +229,6 @@ if (( ${CLEAN} == 1 )); then
     cd ${REPODIR}
 fi
 
-# Before
 
 ################################################################################
 # Configure for building all C++ targets
@@ -283,11 +282,16 @@ fi
 
 # Build and (optionally) install the cuml Python package
 if completeBuild || hasArg cuml || hasArg pydocs; then
+    # Append `-DFIND_CUML_CPP=ON` to CUML_EXTRA_CMAKE_ARGS unless a user specified the option.
+    if [[ "${CUML_EXTRA_CMAKE_ARGS}" != *"DFIND_CUML_CPP"* ]]; then
+        CUML_EXTRA_CMAKE_ARGS="${CUML_EXTRA_CMAKE_ARGS} -DFIND_CUML_CPP=ON"
+    fi
+
     cd ${REPODIR}/python
+
+    python setup.py build_ext --inplace -- -DCMAKE_LIBRARY_PATH=${LIBCUML_BUILD_DIR} -DCMAKE_MESSAGE_LOG_LEVEL=${CMAKE_LOG_LEVEL} ${CUML_EXTRA_CMAKE_ARGS} -- -j${PARALLEL_LEVEL:-1}
     if [[ ${INSTALL_TARGET} != "" ]]; then
-        python setup.py build_ext -j${PARALLEL_LEVEL:-1} ${CUML_EXTRA_PYTHON_ARGS} --library-dir=${LIBCUML_BUILD_DIR} install --single-version-externally-managed --record=record.txt
-    else
-        python setup.py build_ext -j${PARALLEL_LEVEL:-1} ${CUML_EXTRA_PYTHON_ARGS} --library-dir=${LIBCUML_BUILD_DIR}
+        python setup.py install --single-version-externally-managed --record=record.txt  -- -DCMAKE_LIBRARY_PATH=${LIBCUML_BUILD_DIR} -DCMAKE_MESSAGE_LOG_LEVEL=${CMAKE_LOG_LEVEL} ${CUML_EXTRA_CMAKE_ARGS} -- -j${PARALLEL_LEVEL:-1}
     fi
 
     if hasArg pydocs; then

diff --git a/ci/checks/style.sh b/ci/checks/style.sh
@@ -15,7 +15,7 @@ cd "$WORKSPACE"
 
 export GIT_DESCRIBE_TAG=`git describe --tags`
 export MINOR_VERSION=`echo $GIT_DESCRIBE_TAG | grep -o -E '([0-9]+\.[0-9]+)'`
-export UCX_PY_VERSION='0.26.*'
+export UCX_PY_VERSION='0.28.*'
 mamba install "ucx-py=${UCX_PY_VERSION}" "ucx-proc=*=gpu"
 
 # Run flake8 and get results/return code

diff --git a/ci/cpu/build.sh b/ci/cpu/build.sh
@@ -32,7 +32,7 @@ if [ "${IS_STABLE_BUILD}" != "true" ] ; then
 fi
 
 # ucx-py version
-export UCX_PY_VERSION='0.26.*'
+export UCX_PY_VERSION='0.28.*'
 
 ################################################################################
 # SETUP - Check environment

diff --git a/ci/gpu/build.sh b/ci/gpu/build.sh
@@ -33,7 +33,7 @@ export MINOR_VERSION=`echo $GIT_DESCRIBE_TAG | grep -o -E '([0-9]+\.[0-9]+)'`
 unset GIT_DESCRIBE_TAG
 
 # ucx-py version
-export UCX_PY_VERSION='0.26.*'
+export UCX_PY_VERSION='0.28.*'
 
 # configure numba threading library
 export NUMBA_THREADING_LAYER=workqueue
@@ -66,7 +66,7 @@ gpuci_mamba_retry install -c conda-forge -c rapidsai -c rapidsai-nightly -c nvid
       "dask-cuda=${MINOR_VERSION}" \
       "ucx-py=${UCX_PY_VERSION}" \
       "ucx-proc=*=gpu" \
-      "xgboost=1.5.2dev.rapidsai${MINOR_VERSION}" \
+      "xgboost=1.6.0dev.rapidsai${MINOR_VERSION}" \
       "rapids-build-env=${MINOR_VERSION}.*" \
       "rapids-notebook-env=${MINOR_VERSION}.*" \
       "shap>=0.37,<=0.39"
@@ -124,8 +124,8 @@ if [[ -z "$PROJECT_FLASH" || "$PROJECT_FLASH" == "0" ]]; then
 
     gpuci_logger "Install the main version of dask and distributed"
     set -x
-    pip install "git+https://github.com/dask/distributed.git@2022.05.2" --upgrade --no-deps
-    pip install "git+https://github.com/dask/dask.git@2022.05.2" --upgrade --no-deps
+    pip install "git+https://github.com/dask/distributed.git@2022.7.1" --upgrade --no-deps
+    pip install "git+https://github.com/dask/dask.git@2022.7.1" --upgrade --no-deps
     set +x
 
     gpuci_logger "Python pytest for cuml"
@@ -191,14 +191,14 @@ else
     gpuci_logger "Building and installing cuml"
     export CONDA_BLD_DIR="$WORKSPACE/.conda-bld"
     export VERSION_SUFFIX=""
-    gpuci_conda_retry mambabuild --no-build-id --croot ${CONDA_BLD_DIR} conda/recipes/cuml -c ${CONDA_ARTIFACT_PATH} --python=${PYTHON}
-    gpuci_mamba_retry install -c ${CONDA_ARTIFACT_PATH} -c ${CONDA_BLD_DIR} cuml
+    gpuci_conda_retry mambabuild --croot ${CONDA_BLD_DIR} conda/recipes/cuml -c ${CONDA_ARTIFACT_PATH} --python=${PYTHON}
+    gpuci_mamba_retry install cuml -c "${CONDA_BLD_DIR}" -c "${CONDA_ARTIFACT_PATH}"
 
     gpuci_logger "Install the main version of dask, distributed, and dask-glm"
     set -x
 
-    pip install "git+https://github.com/dask/distributed.git@2022.05.2" --upgrade --no-deps
-    pip install "git+https://github.com/dask/dask.git@2022.05.2" --upgrade --no-deps
+    pip install "git+https://github.com/dask/distributed.git@2022.7.1" --upgrade --no-deps
+    pip install "git+https://github.com/dask/dask.git@2022.7.1" --upgrade --no-deps
     pip install "git+https://github.com/dask/dask-glm@main" --force-reinstall --no-deps
     pip install sparse
 

diff --git a/ci/release/update-version.sh b/ci/release/update-version.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-# Copyright (c) 2019-2021, NVIDIA CORPORATION.
+# Copyright (c) 2019-2022, NVIDIA CORPORATION.
 ########################
 # cuML Version Updater #
 ########################
@@ -32,7 +32,10 @@ function sed_runner() {
 }
 
 sed_runner 's/'"CUML VERSION .* LANGUAGES"'/'"CUML VERSION ${NEXT_FULL_TAG} LANGUAGES"'/g' cpp/CMakeLists.txt
-sed_runner 's/'"branch-.*\/RAPIDS.cmake"'/'"branch-${NEXT_SHORT_TAG}\/RAPIDS.cmake"'/g' cpp/CMakeLists.txt
+sed_runner 's/'"set(CUML_VERSION .*)"'/'"set(CUML_VERSION ${NEXT_FULL_TAG})"'/g' python/CMakeLists.txt
+# rapids-cmake version
+sed_runner 's/'"branch-.*\/RAPIDS.cmake"'/'"branch-${NEXT_SHORT_TAG}\/RAPIDS.cmake"'/g' fetch_rapids.cmake
+
 
 # RTD update
 sed_runner 's/version = .*/version = '"'${NEXT_SHORT_TAG}'"'/g' docs/source/conf.py
@@ -44,6 +47,10 @@ for FILE in conda/environments/*.yml; do
    sed_runner "s/dask-cuda=${CURRENT_SHORT_TAG}/dask-cuda=${NEXT_SHORT_TAG}/g" ${FILE};
    sed_runner "s/dask-cudf=${CURRENT_SHORT_TAG}/dask-cudf=${NEXT_SHORT_TAG}/g" ${FILE};
    sed_runner "s/libcumlprims=${CURRENT_SHORT_TAG}/libcumlprims=${NEXT_SHORT_TAG}/g" ${FILE};
+   sed_runner "s/libraft-headers=${CURRENT_SHORT_TAG}/libraft-headers=${NEXT_SHORT_TAG}/g" ${FILE};
+   sed_runner "s/libraft-distance=${CURRENT_SHORT_TAG}/libraft-distance=${NEXT_SHORT_TAG}/g" ${FILE};
+   sed_runner "s/libraft-nn=${CURRENT_SHORT_TAG}/libraft-nn=${NEXT_SHORT_TAG}/g" ${FILE};
+   sed_runner "s/pyraft=${CURRENT_SHORT_TAG}/pyraft=${NEXT_SHORT_TAG}/g" ${FILE};
    sed_runner "s/rapids-build-env=${CURRENT_SHORT_TAG}/rapids-build-env=${NEXT_SHORT_TAG}/g" ${FILE};
    sed_runner "s/rapids-notebook-env=${CURRENT_SHORT_TAG}/rapids-notebook-env=${NEXT_SHORT_TAG}/g" ${FILE};
    sed_runner "s/rapids-doc-env=${CURRENT_SHORT_TAG}/rapids-doc-env=${NEXT_SHORT_TAG}/g" ${FILE};

diff --git a/conda/environments/builddocs_py37.yml b/conda/environments/builddocs_py37.yml
@@ -20,12 +20,11 @@ dependencies:
 # required for building rapids project docs
 - sphinx
 - sphinx_rtd_theme
+- sphinx-markdown-tables
 - sphinxcontrib-websupport
 - nbsphinx
 - numpydoc
 - ipython
 - recommonmark
 - pandoc=<2.0.0
 - pip
-- pip:
-  - sphinx-markdown-tables
diff --git a/conda/environments/cuml_dev_cuda11.0.yml b/conda/environments/cuml_dev_cuda11.0.yml
@@ -6,38 +6,39 @@ channels:
 - conda-forge
 dependencies:
 - cudatoolkit=11.0
-- cuda-python >=11.5,<12.0
-- rapids-build-env=22.06.*
-- rapids-notebook-env=22.06.*
-- rapids-doc-env=22.06.*
-- cudf=22.06.*
-- rmm=22.06.*
-- libcumlprims=22.06.*
-- libraft-headers=22.06.*
-- libraft-distance=22.06.*
-- libraft-nn=22.06.*
-- pyraft=22.06.*
-- dask-cudf=22.06.*
-- dask-cuda=22.06.*
-- ucx>=1.12.1
-- ucx-py=0.26.*
+- cuda-python >=11.5,<11.7.1
+- rapids-build-env=22.10.*
+- rapids-notebook-env=22.10.*
+- rapids-doc-env=22.10.*
+- scikit-build>=0.13.1
+- cudf=22.10.*
+- rmm=22.10.*
+- libcumlprims=22.10.*
+- libraft-headers=22.10.*
+- libraft-distance=22.10.*
+- libraft-nn=22.10.*
+- pyraft=22.10.*
+- dask-cudf=22.10.*
+- dask-cuda=22.10.*
+- ucx>=1.13.0
+- ucx-py=0.28.*
 - ucx-proc=*=gpu
 - dask-ml
 - doxygen>=1.8.20
 - libfaiss>=1.7.0
 - faiss-proc=*=cuda
 - umap-learn
 - scikit-learn=0.24
+- sphinx-markdown-tables
 - treelite=2.4.0
 - statsmodels
 - seaborn
 - hdbscan
 - nltk
 - pip
 - pip:
-    - sphinx_markdown_tables
-    - git+https://github.com/dask/dask.git@2022.05.2
-    - git+https://github.com/dask/distributed.git@2022.05.2
+    - git+https://github.com/dask/dask.git@2022.7.1
+    - git+https://github.com/dask/distributed.git@2022.7.1
     - git+https://github.com/dask/dask-glm@main
     - sparse
 

diff --git a/conda/environments/cuml_dev_cuda11.2.yml b/conda/environments/cuml_dev_cuda11.2.yml
@@ -6,38 +6,39 @@ channels:
 - conda-forge
 dependencies:
 - cudatoolkit=11.2
-- cuda-python >=11.5,<12.0
-- rapids-build-env=22.06.*
-- rapids-notebook-env=22.06.*
-- rapids-doc-env=22.06.*
-- cudf=22.06.*
-- rmm=22.06.*
-- libcumlprims=22.06.*
-- libraft-headers=22.06.*
-- libraft-distance=22.06.*
-- libraft-nn=22.06.*
-- pyraft=22.06.*
-- dask-cudf=22.06.*
-- dask-cuda=22.06.*
-- ucx>=1.12.1
-- ucx-py=0.26.*
+- cuda-python >=11.5,<11.7.1
+- rapids-build-env=22.10.*
+- rapids-notebook-env=22.10.*
+- rapids-doc-env=22.10.*
+- scikit-build>=0.13.1
+- cudf=22.10.*
+- rmm=22.10.*
+- libcumlprims=22.10.*
+- libraft-headers=22.10.*
+- libraft-distance=22.10.*
+- libraft-nn=22.10.*
+- pyraft=22.10.*
+- dask-cudf=22.10.*
+- dask-cuda=22.10.*
+- ucx>=1.13.0
+- ucx-py=0.28.*
 - ucx-proc=*=gpu
 - dask-ml
 - doxygen>=1.8.20
 - libfaiss>=1.7.0
 - faiss-proc=*=cuda
 - umap-learn
 - scikit-learn=0.24
+- sphinx-markdown-tables
 - treelite=2.4.0
 - statsmodels
 - seaborn
 - hdbscan
 - nltk
 - pip
 - pip:
-    - sphinx_markdown_tables
-    - git+https://github.com/dask/dask.git@2022.05.2
-    - git+https://github.com/dask/distributed.git@2022.05.2
+    - git+https://github.com/dask/dask.git@2022.7.1
+    - git+https://github.com/dask/distributed.git@2022.7.1
     - git+https://github.com/dask/dask-glm@main
     - sparse
 

diff --git a/conda/environments/cuml_dev_cuda11.4.yml b/conda/environments/cuml_dev_cuda11.4.yml
@@ -6,38 +6,39 @@ channels:
 - conda-forge
 dependencies:
 - cudatoolkit=11.4
-- cuda-python >=11.5,<12.0
-- rapids-build-env=22.06.*
-- rapids-notebook-env=22.06.*
-- rapids-doc-env=22.06.*
-- cudf=22.06.*
-- rmm=22.06.*
-- libcumlprims=22.06.*
-- libraft-headers=22.06.*
-- libraft-distance=22.06.*
-- libraft-nn=22.06.*
-- pyraft=22.06.*
-- dask-cudf=22.06.*
-- dask-cuda=22.06.*
-- ucx>=1.12.1
-- ucx-py=0.26.*
+- cuda-python >=11.5,<11.7.1
+- rapids-build-env=22.10.*
+- rapids-notebook-env=22.10.*
+- rapids-doc-env=22.10.*
+- scikit-build>=0.13.1
+- cudf=22.10.*
+- rmm=22.10.*
+- libcumlprims=22.10.*
+- libraft-headers=22.10.*
+- libraft-distance=22.10.*
+- libraft-nn=22.10.*
+- pyraft=22.10.*
+- dask-cudf=22.10.*
+- dask-cuda=22.10.*
+- ucx>=1.13.0
+- ucx-py=0.28.*
 - ucx-proc=*=gpu
 - dask-ml
 - doxygen>=1.8.20
 - libfaiss>=1.7.0
 - faiss-proc=*=cuda
 - umap-learn
 - scikit-learn=0.24
+- sphinx-markdown-tables
 - treelite=2.4.0
 - statsmodels
 - seaborn
 - hdbscan
 - nltk
 - pip
 - pip:
-    - sphinx_markdown_tables
-    - git+https://github.com/dask/dask.git@2022.05.2
-    - git+https://github.com/dask/distributed.git@2022.05.2
+    - git+https://github.com/dask/dask.git@2022.7.1
+    - git+https://github.com/dask/distributed.git@2022.7.1
     - git+https://github.com/dask/dask-glm.git@main
     - sparse