Skip to content

Commit 2fd27f5

Browse files
ROCm: Add gfx1150/gfx1151 to build targets
1 parent f6854da commit 2fd27f5

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

.github/scripts/build-rocm.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@ declare rocm_version
66
set -xeuo pipefail
77
bnb_rocm_arch="gfx90a;gfx942;gfx1100;gfx1101"
88

9-
# ROCm 6.4+ - Add gfx1200/gfx1201. Note we assume >=6.4.1.
10-
[[ "${rocm_version}" == 6.4.* || "${rocm_version}" == 7.* ]] && bnb_rocm_arch="${bnb_rocm_arch};gfx1200;gfx1201"
9+
# ROCm 6.4+ - Add gfx1150/gfx1151/gfx1200/gfx1201. Note we assume >=6.4.4.
10+
[[ "${rocm_version}" == 6.4.* || "${rocm_version}" == 7.* ]] && bnb_rocm_arch="${bnb_rocm_arch};gfx1150;gfx1151;gfx1200;gfx1201"
1111

1212
# ROCm 7.0+ - Add gfx950
1313
[[ "${rocm_version}" == 7.* ]] && bnb_rocm_arch="${bnb_rocm_arch};gfx950"

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ bitsandbytes has the following minimum requirements for all platforms:
7272
<td>🟥 AMD GPU <br><code>cuda</code></td>
7373
<td>
7474
CDNA: gfx90a, gfx942, gfx950<br>
75-
RDNA: gfx1100, gfx1200, gfx1201
75+
RDNA: gfx1100, gfx1150, gfx1151, gfx1200, gfx1201
7676
</td>
7777
<td>✅</td>
7878
<td>〰️</td>

docs/source/installation.mdx

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -194,9 +194,9 @@ The currently distributed preview `bitsandbytes` are built with the following co
194194
|--------------------|----------|---------------------------------------------------------------------|
195195
| **Linux x86-64** | 6.2.4 | CDNA: gfx90a, gfx942 / RDNA: gfx1100, gfx1101
196196
| **Linux x86-64** | 6.3.4 | CDNA: gfx90a, gfx942 / RDNA: gfx1100, gfx1101
197-
| **Linux x86-64** | 6.4.4 | CDNA: gfx90a, gfx942 / RDNA: gfx1100, gfx1101, gfx1200, gfx1201
198-
| **Linux x86-64** | 7.0.2 | CDNA: gfx90a, gfx942, gfx950 / RDNA: gfx1100 / gfx1101 / gfx1200 / gfx1201
199-
| **Linux x86-64** | 7.1.0 | CDNA: gfx90a, gfx942, gfx950 / RDNA: gfx1100 / gfx1101 / gfx1200 / gfx1201
197+
| **Linux x86-64** | 6.4.4 | CDNA: gfx90a, gfx942 / RDNA: gfx1100, gfx1101, gfx1150, gfx1151, gfx1200, gfx1201
198+
| **Linux x86-64** | 7.0.2 | CDNA: gfx90a, gfx942, gfx950 / RDNA: gfx1100, gfx1101, gfx1150, gfx1151, gfx1200, gfx1201
199+
| **Linux x86-64** | 7.1.0 | CDNA: gfx90a, gfx942, gfx950 / RDNA: gfx1100, gfx1101, gfx1150, gfx1151, gfx1200, gfx1201
200200

201201
**Windows is not currently supported.**
202202

0 commit comments

Comments
 (0)