From 49dcd4a3e522abe077af6f17ccfd919caf499a65 Mon Sep 17 00:00:00 2001 From: Ruihang Lai Date: Mon, 16 Dec 2024 13:23:04 +0800 Subject: [PATCH] [CI] Update TVM package for CI (#3067) This PR updates the TVM package used for CI from cuda 12.1 to 12.3. --- ci/task/test_model_compile.sh | 2 +- ci/task/test_unittest.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ci/task/test_model_compile.sh b/ci/task/test_model_compile.sh index d5064e7b92..40a00df9d6 100755 --- a/ci/task/test_model_compile.sh +++ b/ci/task/test_model_compile.sh @@ -9,7 +9,7 @@ pip install --force-reinstall wheels/*.whl if [[ ${GPU} == cuda* ]]; then TARGET=cuda - pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu121 + pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu123 export LD_LIBRARY_PATH=/usr/local/cuda/compat/:$LD_LIBRARY_PATH elif [[ ${GPU} == rocm* ]]; then TARGET=rocm diff --git a/ci/task/test_unittest.sh b/ci/task/test_unittest.sh index fdc02dd46a..bb459e5e10 100755 --- a/ci/task/test_unittest.sh +++ b/ci/task/test_unittest.sh @@ -8,7 +8,7 @@ if [[ -n ${MLC_CI_SETUP_DEPS:-} ]]; then # Install dependency pip install --force-reinstall wheels/*.whl pip install --quiet pytest - pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu121 + pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu123 export LD_LIBRARY_PATH=/usr/local/cuda/compat/:$LD_LIBRARY_PATH fi