Skip to content

Commit 49dcd4a

Browse files
authored
[CI] Update TVM package for CI (#3067)
This PR updates the TVM package used for CI from cuda 12.1 to 12.3.
1 parent 9dd0788 commit 49dcd4a

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

ci/task/test_model_compile.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ pip install --force-reinstall wheels/*.whl
99

1010
if [[ ${GPU} == cuda* ]]; then
1111
TARGET=cuda
12-
pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu121
12+
pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu123
1313
export LD_LIBRARY_PATH=/usr/local/cuda/compat/:$LD_LIBRARY_PATH
1414
elif [[ ${GPU} == rocm* ]]; then
1515
TARGET=rocm

ci/task/test_unittest.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ if [[ -n ${MLC_CI_SETUP_DEPS:-} ]]; then
88
# Install dependency
99
pip install --force-reinstall wheels/*.whl
1010
pip install --quiet pytest
11-
pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu121
11+
pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cu123
1212
export LD_LIBRARY_PATH=/usr/local/cuda/compat/:$LD_LIBRARY_PATH
1313
fi
1414

0 commit comments

Comments
 (0)