Add rocm arch to compile command

bitsandbytes-foundation · Jul 30, 2024 · b78b340 · b78b340
1 parent 7c91909
commit b78b340
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/scripts/build-rocm.sh b/.github/scripts/build-rocm.sh
@@ -4,14 +4,15 @@ declare build_os
 declare rocm_version
 
 set -xeuo pipefail
+bnb_rocm_arch="gfx906;gfx908;gfx90a;gfx942;gfx1100;gfx1030"
 if [ "${build_os:0:6}" == ubuntu ]; then
 	image=rocm/dev-ubuntu-22.04:${rocm_version}-complete
 	echo "Using image $image"
 	docker run --rm --platform "linux/$build_arch" -i \
 		-w /src -v "$PWD:/src" "$image" sh -c \
 		"apt-get update \
       && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends cmake \
-      && cmake -DCOMPUTE_BACKEND=hip . \
+      && cmake -DCOMPUTE_BACKEND=hip -DBNB_ROCM_ARCH=\"${bnb_rocm_arch}\" . \
       && cmake --build ."
 fi