From 8eca4f98445f19bb5189871c84c8b92e82900321 Mon Sep 17 00:00:00 2001 From: sstamenk Date: Wed, 4 Mar 2026 14:22:36 +0100 Subject: [PATCH] Enable more targets for ROCm --- .github/scripts/build-rocm.sh | 6 +++--- CMakeLists.txt | 2 +- README.md | 2 +- docs/source/installation.mdx | 12 ++++++------ 4 files changed, 11 insertions(+), 11 deletions(-) diff --git a/.github/scripts/build-rocm.sh b/.github/scripts/build-rocm.sh index d0efa5bb3..dd44beb5b 100644 --- a/.github/scripts/build-rocm.sh +++ b/.github/scripts/build-rocm.sh @@ -4,10 +4,10 @@ declare build_os declare rocm_version set -xeuo pipefail -bnb_rocm_arch="gfx90a;gfx942;gfx1100;gfx1101" +bnb_rocm_arch="gfx90a;gfx942;gfx1100;gfx1101;gfx1102;gfx1103" -# ROCm 6.4+ - Add gfx1150/gfx1151/gfx1200/gfx1201. Note we assume >=6.4.4. -[[ "${rocm_version}" == 6.4.* || "${rocm_version}" == 7.* ]] && bnb_rocm_arch="${bnb_rocm_arch};gfx1150;gfx1151;gfx1200;gfx1201" +# ROCm 6.4+ - Add RDNA4 and RDNA3.5 targets. Note we assume >=6.4.4. +[[ "${rocm_version}" == 6.4.* || "${rocm_version}" == 7.* ]] && bnb_rocm_arch="${bnb_rocm_arch};gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201" # ROCm 7.0+ - Add gfx950 [[ "${rocm_version}" == 7.* ]] && bnb_rocm_arch="${bnb_rocm_arch};gfx950" diff --git a/CMakeLists.txt b/CMakeLists.txt index 81326ffcf..952d5d4ea 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -236,7 +236,7 @@ elseif(BUILD_HIP) elseif(AMDGPU_TARGETS AND NOT CMAKE_HIP_ARCHITECTURES) set(CMAKE_HIP_ARCHITECTURES ${AMDGPU_TARGETS}) elseif(NOT CMAKE_HIP_ARCHITECTURES) - set(CMAKE_HIP_ARCHITECTURES "gfx90a;gfx942;gfx1100;gfx1101;gfx1150;gfx1151;gfx1200;gfx1201") + set(CMAKE_HIP_ARCHITECTURES "gfx90a;gfx942;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201") endif() enable_language(HIP) diff --git a/README.md b/README.md index d3d74efc8..caf53d5f9 100644 --- a/README.md +++ b/README.md @@ -72,7 +72,7 @@ bitsandbytes has the following minimum requirements for all platforms: 🟥 AMD GPU
cuda CDNA: gfx90a, gfx942, gfx950
- RDNA: gfx1100, gfx1101, gfx1150, gfx1151, gfx1200, gfx1201 + RDNA: gfx1100, gfx1101, gfx1102, gfx1103, gfx1150, gfx1151, gfx1152, gfx1153, gfx1200, gfx1201 ✅ ✅ diff --git a/docs/source/installation.mdx b/docs/source/installation.mdx index 3d170436d..c97996b75 100644 --- a/docs/source/installation.mdx +++ b/docs/source/installation.mdx @@ -197,12 +197,12 @@ The currently distributed `bitsandbytes` are built with the following configurat | **OS** | **ROCm** | **Targets** |--------------------|----------|---------------------------------------------------------------------| -| **Linux x86-64** | 6.2.4 | CDNA: gfx90a, gfx942 / RDNA: gfx1100, gfx1101 -| **Linux x86-64** | 6.3.4 | CDNA: gfx90a, gfx942 / RDNA: gfx1100, gfx1101 -| **Linux x86-64** | 6.4.4 | CDNA: gfx90a, gfx942 / RDNA: gfx1100, gfx1101, gfx1150, gfx1151, gfx1200, gfx1201 -| **Linux x86-64** | 7.0.2 | CDNA: gfx90a, gfx942, gfx950 / RDNA: gfx1100, gfx1101, gfx1150, gfx1151, gfx1200, gfx1201 -| **Linux x86-64** | 7.1.0 | CDNA: gfx90a, gfx942, gfx950 / RDNA: gfx1100, gfx1101, gfx1150, gfx1151, gfx1200, gfx1201 -| **Linux x86-64** | 7.2.0 | CDNA: gfx90a, gfx942, gfx950 / RDNA: gfx1100, gfx1101, gfx1150, gfx1151, gfx1200, gfx1201 +| **Linux x86-64** | 6.2.4 | CDNA: gfx90a, gfx942 / RDNA: gfx1100, gfx1101, gfx1102, gfx1103 +| **Linux x86-64** | 6.3.4 | CDNA: gfx90a, gfx942 / RDNA: gfx1100, gfx1101, gfx1102, gfx1103 +| **Linux x86-64** | 6.4.4 | CDNA: gfx90a, gfx942 / RDNA: gfx1100, gfx1101, gfx1102, gfx1103, gfx1150, gfx1151, gfx1152, gfx1153, gfx1200, gfx1201 +| **Linux x86-64** | 7.0.2 | CDNA: gfx90a, gfx942, gfx950 / RDNA: gfx1100, gfx1101, gfx1102, gfx1103, gfx1150, gfx1151, gfx1152, gfx1153, gfx1200, gfx1201 +| **Linux x86-64** | 7.1.0 | CDNA: gfx90a, gfx942, gfx950 / RDNA: gfx1100, gfx1101, gfx1102, gfx1103, gfx1150, gfx1151, gfx1152, gfx1153, gfx1200, gfx1201 +| **Linux x86-64** | 7.2.0 | CDNA: gfx90a, gfx942, gfx950 / RDNA: gfx1100, gfx1101, gfx1102, gfx1103, gfx1150, gfx1151, gfx1152, gfx1153, gfx1200, gfx1201 **Windows is not currently supported.**