diff --git a/cmake/thirdparty/get_cuml.cmake b/cmake/thirdparty/get_cuml.cmake index 2f02c88..bbd1285 100644 --- a/cmake/thirdparty/get_cuml.cmake +++ b/cmake/thirdparty/get_cuml.cmake @@ -55,7 +55,7 @@ endfunction() # Change pinned tag here to test a commit in CI # To use a different RAFT locally, set the CMake variable # CPM_raft_SOURCE=/path/to/local/raft -find_and_configure_cuml(VERSION ${RAPIDS_TRITON_MIN_VERSION_rapids_projects} +find_and_configure_cuml(VERSION 24.06 FORK rapidsai PINNED_TAG branch-24.06 USE_TREELITE_STATIC ${TRITON_FIL_USE_TREELITE_STATIC} diff --git a/ops/Dockerfile b/ops/Dockerfile index 0a2de45..a7e0ee3 100644 --- a/ops/Dockerfile +++ b/ops/Dockerfile @@ -3,7 +3,7 @@ # Arguments for controlling build details ########################################################################################### # Version of Triton to use -ARG TRITON_VERSION=24.06 +ARG TRITON_VERSION=24.07 # Base container image ARG BASE_IMAGE=nvcr.io/nvidia/tritonserver:${TRITON_VERSION}-py3 # Whether or not to enable GPU build