From 70f97c5a100a11776d10fb8753464c30cdf60a78 Mon Sep 17 00:00:00 2001 From: Philip Hyunsu Cho Date: Wed, 22 Sep 2021 18:00:16 -0700 Subject: [PATCH] Upgrade Treelite to 2.1.0 (#4220) The 2.1.0 version of Treelite incorporates the following major improvements: * dmlc/treelite#311 * dmlc/treelite#302 * dmlc/treelite#303 * dmlc/treelite#296 In particular, dmlc/treelite#311 is a critical follow-up to #4191 and addresses a performance regression. Requires https://github.com/rapidsai/integration/pull/353 Authors: - Philip Hyunsu Cho (https://github.com/hcho3) Approvers: - Jordan Jacobelli (https://github.com/Ethyling) - Dante Gama Dessavre (https://github.com/dantegd) URL: https://github.com/rapidsai/cuml/pull/4220 --- conda/environments/cuml_dev_cuda11.0.yml | 2 +- conda/environments/cuml_dev_cuda11.2.yml | 2 +- conda/environments/cuml_dev_cuda11.4.yml | 2 +- conda/recipes/cuml/meta.yaml | 4 ++-- conda/recipes/libcuml/meta.yaml | 4 ++-- cpp/cmake/thirdparty/get_treelite.cmake | 4 ++-- 6 files changed, 9 insertions(+), 9 deletions(-) diff --git a/conda/environments/cuml_dev_cuda11.0.yml b/conda/environments/cuml_dev_cuda11.0.yml index 320a7cb43a..9fdecefa21 100644 --- a/conda/environments/cuml_dev_cuda11.0.yml +++ b/conda/environments/cuml_dev_cuda11.0.yml @@ -22,7 +22,7 @@ dependencies: - faiss-proc=*=cuda - umap-learn - scikit-learn=0.23.1 -- treelite=2.0.0 +- treelite=2.1.0 - statsmodels - seaborn - hdbscan diff --git a/conda/environments/cuml_dev_cuda11.2.yml b/conda/environments/cuml_dev_cuda11.2.yml index 7ca5f6bd11..7759958426 100644 --- a/conda/environments/cuml_dev_cuda11.2.yml +++ b/conda/environments/cuml_dev_cuda11.2.yml @@ -22,7 +22,7 @@ dependencies: - faiss-proc=*=cuda - umap-learn - scikit-learn=0.23.1 -- treelite=2.0.0 +- treelite=2.1.0 - statsmodels - seaborn - hdbscan diff --git a/conda/environments/cuml_dev_cuda11.4.yml b/conda/environments/cuml_dev_cuda11.4.yml index e5a637cb1e..57505c9e78 100644 --- a/conda/environments/cuml_dev_cuda11.4.yml +++ b/conda/environments/cuml_dev_cuda11.4.yml @@ -22,7 +22,7 @@ dependencies: - faiss-proc=*=cuda - umap-learn - scikit-learn=0.23.1 -- treelite=2.0.0 +- treelite=2.1.0 - statsmodels - seaborn - hdbscan diff --git a/conda/recipes/cuml/meta.yaml b/conda/recipes/cuml/meta.yaml index 485d65d6ea..1eadec7d64 100644 --- a/conda/recipes/cuml/meta.yaml +++ b/conda/recipes/cuml/meta.yaml @@ -28,7 +28,7 @@ requirements: - setuptools - cython>=0.29,<0.30 - cmake>=3.20.1 - - treelite=2.0.0 + - treelite=2.1.0 - cudf {{ minor_version }} - libcuml={{ version }} - libcumlprims {{ minor_version }} @@ -42,7 +42,7 @@ requirements: - libcuml={{ version }} - libcumlprims {{ minor_version }} - cupy>=7.8.0,<10.0.0a0 - - treelite=2.0.0 + - treelite=2.1.0 - nccl>=2.9.9 - ucx-py 0.22 - ucx-proc=*=gpu diff --git a/conda/recipes/libcuml/meta.yaml b/conda/recipes/libcuml/meta.yaml index 644d7f22ed..1297ca05d2 100644 --- a/conda/recipes/libcuml/meta.yaml +++ b/conda/recipes/libcuml/meta.yaml @@ -43,7 +43,7 @@ requirements: - ucx-proc=*=gpu - libcumlprims {{ minor_version }} - lapack - - treelite=2.0.0 + - treelite=2.1.0 - faiss-proc=*=cuda - gtest=1.10.0 - gmock @@ -55,7 +55,7 @@ requirements: - ucx-py 0.22 - ucx-proc=*=gpu - {{ pin_compatible('cudatoolkit', max_pin='x.x') }} - - treelite=2.0.0 + - treelite=2.1.0 - faiss-proc=*=cuda - libfaiss 1.7.0 *_cuda diff --git a/cpp/cmake/thirdparty/get_treelite.cmake b/cpp/cmake/thirdparty/get_treelite.cmake index adf7958c7f..171706ea20 100644 --- a/cpp/cmake/thirdparty/get_treelite.cmake +++ b/cpp/cmake/thirdparty/get_treelite.cmake @@ -54,5 +54,5 @@ function(find_and_configure_treelite) endfunction() -find_and_configure_treelite(VERSION 2.0.0 - PINNED_TAG b117da58d7d9a5cc54aa3711e5ad9a8407734c6e) +find_and_configure_treelite(VERSION 2.1.0 + PINNED_TAG e5248931c62e3807248e0b150e27b2530a510634)