diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml index bf73b938a5..401aa4121c 100644 --- a/conda/environments/all_cuda-118_arch-x86_64.yaml +++ b/conda/environments/all_cuda-118_arch-x86_64.yaml @@ -76,7 +76,7 @@ dependencies: - sphinx-markdown-tables - statsmodels - sysroot_linux-64==2.17 -- treelite==4.3.0 +- treelite==4.4.1 - umap-learn==0.5.6 - xgboost>=2.1.0 name: all_cuda-118_arch-x86_64 diff --git a/conda/environments/all_cuda-125_arch-x86_64.yaml b/conda/environments/all_cuda-125_arch-x86_64.yaml index 72539f2d18..23badefc90 100644 --- a/conda/environments/all_cuda-125_arch-x86_64.yaml +++ b/conda/environments/all_cuda-125_arch-x86_64.yaml @@ -72,7 +72,7 @@ dependencies: - sphinx-markdown-tables - statsmodels - sysroot_linux-64==2.17 -- treelite==4.3.0 +- treelite==4.4.1 - umap-learn==0.5.6 - xgboost>=2.1.0 name: all_cuda-125_arch-x86_64 diff --git a/conda/recipes/cuml/conda_build_config.yaml b/conda/recipes/cuml/conda_build_config.yaml index eb829ba4de..0dde68a976 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.3.0" + - "=4.4.1" diff --git a/conda/recipes/libcuml/conda_build_config.yaml b/conda/recipes/libcuml/conda_build_config.yaml index f802440354..5ac453c0ea 100644 --- a/conda/recipes/libcuml/conda_build_config.yaml +++ b/conda/recipes/libcuml/conda_build_config.yaml @@ -26,7 +26,7 @@ spdlog_version: - ">=1.14.1,<1.15" treelite_version: - - "=4.3.0" + - "=4.4.1" # 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 a14bacb531..9a221f1b85 100644 --- a/cpp/cmake/thirdparty/get_treelite.cmake +++ b/cpp/cmake/thirdparty/get_treelite.cmake @@ -1,5 +1,5 @@ #============================================================================= -# Copyright (c) 2021-2024, NVIDIA CORPORATION. +# Copyright (c) 2021-2025, NVIDIA CORPORATION. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -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.3.0 - PINNED_TAG 575e4208f2b18e40d818c338ecb95d7a26e69aab +find_and_configure_treelite(VERSION 4.4.1 + PINNED_TAG 386bd0de99f5a66584c7e58221ee38ce606ad1ae EXCLUDE_FROM_ALL ${CUML_EXCLUDE_TREELITE_FROM_ALL} BUILD_STATIC_LIBS ${CUML_USE_TREELITE_STATIC}) diff --git a/dependencies.yaml b/dependencies.yaml index 90e6dc5fc2..91e7d0623d 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -215,7 +215,7 @@ dependencies: - output_types: [conda, requirements, pyproject] packages: - &cython cython>=3.0.0 - - &treelite treelite==4.3.0 + - &treelite treelite==4.4.1 specific: - output_types: [conda, requirements, pyproject] matrices: diff --git a/python/cuml/pyproject.toml b/python/cuml/pyproject.toml index 2f0521fe6e..95885de503 100644 --- a/python/cuml/pyproject.toml +++ b/python/cuml/pyproject.toml @@ -109,7 +109,7 @@ dependencies = [ "rapids-dask-dependency==25.2.*,>=0.0.0a0", "rmm==25.2.*,>=0.0.0a0", "scipy>=1.8.0", - "treelite==4.3.0", + "treelite==4.4.1", ] # 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", @@ -181,7 +181,7 @@ requires = [ "ninja", "pylibraft==25.2.*,>=0.0.0a0", "rmm==25.2.*,>=0.0.0a0", - "treelite==4.3.0", + "treelite==4.4.1", ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. [tool.scikit-build]