diff --git a/CMakeLists.txt b/CMakeLists.txt index 6909cdc..52dc314 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -76,7 +76,7 @@ else() ############################################################################## # - Prepare rapids-cmake ----------------------------------------------------- file(DOWNLOAD - https://raw.githubusercontent.com/rapidsai/rapids-cmake/branch-24.04/RAPIDS.cmake + https://raw.githubusercontent.com/rapidsai/rapids-cmake/branch-24.10/RAPIDS.cmake ${CMAKE_BINARY_DIR}/RAPIDS.cmake) include(${CMAKE_BINARY_DIR}/RAPIDS.cmake) include(rapids-cmake) @@ -97,7 +97,7 @@ else() set(BACKEND_FOLDER "/opt/tritonserver/backends" CACHE STRING "Triton backend folder path") # Specify *minimum* version for all RAPIDS dependencies # Some RAPIDS deps may have later versions - set(RAPIDS_DEPENDENCIES_VERSION "24.04" CACHE STRING "RAPIDS projects dependencies version") + set(RAPIDS_DEPENDENCIES_VERSION "24.10" CACHE STRING "RAPIDS projects dependencies version") option(TRITON_FIL_USE_TREELITE_STATIC "Link Treelite statically in libtriton_fil.so and cuml++.so" ON) diff --git a/conda/environments/triton_benchmark.yml b/conda/environments/triton_benchmark.yml index d1a4966..6f193ae 100644 --- a/conda/environments/triton_benchmark.yml +++ b/conda/environments/triton_benchmark.yml @@ -15,4 +15,4 @@ dependencies: - pip: - tritonclient[all] - protobuf - - git+https://github.com/rapidsai/rapids-triton.git@branch-24.04#subdirectory=python + - git+https://github.com/rapidsai/rapids-triton.git@branch-24.10#subdirectory=python diff --git a/conda/environments/triton_test.yml b/conda/environments/triton_test.yml index 85d4a0b..cae091e 100644 --- a/conda/environments/triton_test.yml +++ b/conda/environments/triton_test.yml @@ -22,4 +22,4 @@ dependencies: - pip: - tritonclient[all] - protobuf - - git+https://github.com/rapidsai/rapids-triton.git@branch-24.04#subdirectory=python + - git+https://github.com/rapidsai/rapids-triton.git@branch-24.10#subdirectory=python diff --git a/ops/Dockerfile b/ops/Dockerfile index 31ff4d3..0af8948 100644 --- a/ops/Dockerfile +++ b/ops/Dockerfile @@ -63,7 +63,7 @@ RUN conda run --no-capture-output -n triton_test \ FROM wheel-install-${USE_CLIENT_WHEEL} as conda-test RUN conda run --no-capture-output -n triton_test \ - pip install git+https://github.com/rapidsai/rapids-triton.git@branch-24.04#subdirectory=python + pip install git+https://github.com/rapidsai/rapids-triton.git@branch-24.10#subdirectory=python RUN conda-pack --ignore-missing-files -n triton_test -o /tmp/env.tar \ && mkdir /conda/test/ \ && cd /conda/test/ \ @@ -150,7 +150,7 @@ ENV TRITON_ENABLE_GPU=$TRITON_ENABLE_GPU # Specify *minimum* version for all RAPIDS dependencies # Some RAPIDS deps may have later versions -ARG RAPIDS_DEPENDENCIES_VERSION=24.04 +ARG RAPIDS_DEPENDENCIES_VERSION=24.10 ENV RAPIDS_DEPENDENCIES_VERSION=$RAPIDS_DEPENDENCIES_VERSION ARG TRITON_FIL_USE_TREELITE_STATIC=ON