Bump Treelite to 4.3.0 (#5968)

Treelite 4.3.0 contains the following improvements: * Support XGBoost 2.1.0, including the UBJSON format (dmlc/treelite#572, dmlc/treelite#578) * [GTIL] Allow inferencing with FP32 input + FP64 model (dmlc/treelite#574). Related: triton-inference-server/fil_backend#391 * Prevent integer overflow for deep LightGBM trees by using DFS order (dmlc/treelite#570). * Support building with latest RapidJSON (dmlc/treelite#567) Authors: - Philip Hyunsu Cho (https://github.com/hcho3) Approvers: - James Lamb (https://github.com/jameslamb) - Dante Gama Dessavre (https://github.com/dantegd) URL: #5968
rapidsai · Jul 24, 2024 · d0fcdce · d0fcdce
1 parent 6e5670b
commit d0fcdce
Show file tree

Hide file tree

Showing 7 changed files with 9 additions and 9 deletions.
diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml
@@ -72,7 +72,7 @@ dependencies:
 - sphinx<6
 - statsmodels
 - sysroot_linux-64==2.17
-- treelite==4.2.1
+- treelite==4.3.0
 - umap-learn==0.5.3
 - pip:
   - dask-glm==0.3.0

diff --git a/conda/environments/all_cuda-125_arch-x86_64.yaml b/conda/environments/all_cuda-125_arch-x86_64.yaml
@@ -68,7 +68,7 @@ dependencies:
 - sphinx<6
 - statsmodels
 - sysroot_linux-64==2.17
-- treelite==4.2.1
+- treelite==4.3.0
 - umap-learn==0.5.3
 - pip:
   - dask-glm==0.3.0

diff --git a/conda/recipes/cuml/conda_build_config.yaml b/conda/recipes/cuml/conda_build_config.yaml
@@ -20,4 +20,4 @@ c_stdlib_version:
   - "=2.17"
 
 treelite_version:
-  - "=4.2.1"
+  - "=4.3.0"
diff --git a/conda/recipes/libcuml/conda_build_config.yaml b/conda/recipes/libcuml/conda_build_config.yaml
@@ -20,7 +20,7 @@ cmake_version:
   - ">=3.26.4,!=3.30.0"
 
 treelite_version:
-  - "=4.2.1"
+  - "=4.3.0"
 
 # The CTK libraries below are missing from the conda-forge::cudatoolkit package
 # for CUDA 11. The "*_host_*" version specifiers correspond to `11.8` packages

diff --git a/cpp/cmake/thirdparty/get_treelite.cmake b/cpp/cmake/thirdparty/get_treelite.cmake
@@ -78,7 +78,7 @@ function(find_and_configure_treelite)
     rapids_export_find_package_root(BUILD Treelite [=[${CMAKE_CURRENT_LIST_DIR}]=] EXPORT_SET cuml-exports)
 endfunction()
 
-find_and_configure_treelite(VERSION     4.2.1
-                        PINNED_TAG  07405e7b8d79e84f7e7ab29bb8b00de20dc6aed4
+find_and_configure_treelite(VERSION     4.3.0
+                        PINNED_TAG  575e4208f2b18e40d818c338ecb95d7a26e69aab
                         EXCLUDE_FROM_ALL  ${CUML_EXCLUDE_TREELITE_FROM_ALL}
                         BUILD_STATIC_LIBS ${CUML_USE_TREELITE_STATIC})
diff --git a/dependencies.yaml b/dependencies.yaml
@@ -177,7 +177,7 @@ dependencies:
       - output_types: [conda, requirements, pyproject]
         packages:
           - cython>=3.0.0
-          - &treelite treelite==4.2.1
+          - &treelite treelite==4.3.0
       - output_types: conda
         packages:
           - &pylibraft_conda pylibraft==24.8.*,>=0.0.0a0

diff --git a/python/cuml/pyproject.toml b/python/cuml/pyproject.toml
@@ -92,7 +92,7 @@ dependencies = [
     "rapids-dask-dependency==24.8.*,>=0.0.0a0",
     "rmm==24.8.*,>=0.0.0a0",
     "scipy>=1.8.0",
-    "treelite==4.2.1",
+    "treelite==4.3.0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 classifiers = [
     "Intended Audience :: Developers",
@@ -160,7 +160,7 @@ requires = [
     "ninja",
     "pylibraft==24.8.*,>=0.0.0a0",
     "rmm==24.8.*,>=0.0.0a0",
-    "treelite==4.2.1",
+    "treelite==4.3.0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 
 [tool.scikit-build]