Skip to content

Commit 20ea209

Browse files
Use newer CMake from PyPI for ROCm build
1 parent fd63b2b commit 20ea209

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.github/scripts/build-rocm.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ if [ "${build_os:0:6}" == ubuntu ]; then
1818
docker run --rm --platform "linux/$build_arch" -i \
1919
-w /src -v "$PWD:/src" "$image" sh -c \
2020
"apt-get update \
21-
&& DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends cmake \
21+
&& pip install cmake==3.31.6 \
2222
&& cmake -DCOMPUTE_BACKEND=hip -DBNB_ROCM_ARCH=\"${bnb_rocm_arch}\" . \
2323
&& cmake --build ."
2424
fi

0 commit comments

Comments
 (0)