From d0fcdceb3171bc908221a8d809e8c1af3c945cf0 Mon Sep 17 00:00:00 2001 From: Philip Hyunsu Cho Date: Wed, 24 Jul 2024 14:25:09 -0700 Subject: [PATCH] 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: https://github.com/triton-inference-server/fil_backend/issues/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: https://github.com/rapidsai/cuml/pull/5968 --- conda/environments/all_cuda-118_arch-x86_64.yaml | 2 +- conda/environments/all_cuda-125_arch-x86_64.yaml | 2 +- conda/recipes/cuml/conda_build_config.yaml | 2 +- conda/recipes/libcuml/conda_build_config.yaml | 2 +- cpp/cmake/thirdparty/get_treelite.cmake | 4 ++-- dependencies.yaml | 2 +- python/cuml/pyproject.toml | 4 ++-- 7 files changed, 9 insertions(+), 9 deletions(-) diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml index b912d10d4a..6338d3cac2 100644 --- 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 index 9d1fd86bf9..ca6bf10588 100644 --- 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 index 4a25c94240..eb829ba4de 100644 --- 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 index 9b53073456..42b1cf2e86 100644 --- 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 index 86a2ddf361..a14bacb531 100644 --- 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 index c35d12bb75..53128a885a 100644 --- 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 index 391aa7c7f5..0d8554854b 100644 --- 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]