Upgrade Treelite to 2.4.0 (rapidsai#4752)

The 2.4.0 version of Treelite incorporates the following improvements: * dmlc/treelite#370 * dmlc/treelite#381 * dmlc/treelite#380 * dmlc/treelite#383 Requires rapidsai/integration#474 This PR targets the 22.06 release. Closes rapidsai#4715 Authors: - Philip Hyunsu Cho (https://github.com/hcho3) Approvers: - AJ Schmidt (https://github.com/ajschmidt8) - Corey J. Nolet (https://github.com/cjnolet) URL: rapidsai#4752
vimarsh6739 · May 28, 2022 · bcac0ba · bcac0ba
1 parent 7c745e7
commit bcac0ba
Show file tree

Hide file tree

Showing 7 changed files with 9 additions and 9 deletions.
diff --git a/conda/environments/cuml_dev_cuda11.0.yml b/conda/environments/cuml_dev_cuda11.0.yml
@@ -28,7 +28,7 @@ dependencies:
 - faiss-proc=*=cuda
 - umap-learn
 - scikit-learn=0.24
-- treelite=2.3.0
+- treelite=2.4.0
 - statsmodels
 - seaborn
 - hdbscan

diff --git a/conda/environments/cuml_dev_cuda11.2.yml b/conda/environments/cuml_dev_cuda11.2.yml
@@ -28,7 +28,7 @@ dependencies:
 - faiss-proc=*=cuda
 - umap-learn
 - scikit-learn=0.24
-- treelite=2.3.0
+- treelite=2.4.0
 - statsmodels
 - seaborn
 - hdbscan

diff --git a/conda/environments/cuml_dev_cuda11.4.yml b/conda/environments/cuml_dev_cuda11.4.yml
@@ -28,7 +28,7 @@ dependencies:
 - faiss-proc=*=cuda
 - umap-learn
 - scikit-learn=0.24
-- treelite=2.3.0
+- treelite=2.4.0
 - statsmodels
 - seaborn
 - hdbscan

diff --git a/conda/environments/cuml_dev_cuda11.5.yml b/conda/environments/cuml_dev_cuda11.5.yml
@@ -28,7 +28,7 @@ dependencies:
 - faiss-proc=*=cuda
 - umap-learn
 - scikit-learn=0.24
-- treelite=2.3.0
+- treelite=2.4.0
 - statsmodels
 - seaborn
 - hdbscan

diff --git a/conda/recipes/cuml/meta.yaml b/conda/recipes/cuml/meta.yaml
@@ -35,7 +35,7 @@ requirements:
     - python x.x
     - setuptools
     - cython>=0.29,<0.30
-    - treelite=2.3.0
+    - treelite=2.4.0
     - cudf {{ minor_version }}
     - libcuml={{ version }}
     - libcumlprims {{ minor_version }}
@@ -52,7 +52,7 @@ requirements:
     - libcumlprims {{ minor_version }}
     - pyraft {{ minor_version }}
     - cupy>=7.8.0,<11.0.0a0
-    - treelite=2.3.0
+    - treelite=2.4.0
     - nccl>=2.9.9
     - ucx-py {{ ucx_py_version }}
     - ucx-proc=*=gpu

diff --git a/conda/recipes/libcuml/conda_build_config.yaml b/conda/recipes/libcuml/conda_build_config.yaml
@@ -20,7 +20,7 @@ ucx_version:
   - "1.12.1"
 
 treelite_version:
-  - "2.3.0"
+  - "2.4.0"
 
 gtest_version:
   - "1.10.0"

diff --git a/cpp/cmake/thirdparty/get_treelite.cmake b/cpp/cmake/thirdparty/get_treelite.cmake
@@ -88,6 +88,6 @@ function(find_and_configure_treelite)
     rapids_export_find_package_root(BUILD Treelite [=[${CMAKE_CURRENT_LIST_DIR}]=] cuml-exports)
 endfunction()
 
-find_and_configure_treelite(VERSION     2.3.0
-                        PINNED_TAG  b65bedceef9893c1ad92977dd466f582540ba7ab
+find_and_configure_treelite(VERSION     2.4.0
+                        PINNED_TAG  dcd54779ce9324c69452db9b906f2d258374d5b9
                         BUILD_STATIC_LIBS ${CUML_USE_TREELITE_STATIC})