Skip to content

Commit 2db704e

Browse files
authored
Update branch change instructions
1 parent df97df2 commit 2db704e

1 file changed

Lines changed: 3 additions & 4 deletions

File tree

README.md

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,18 +6,17 @@ The `bitsandbytes` library is a lightweight Python wrapper around CUDA custom fu
66

77
The library includes quantization primitives for 8-bit & 4-bit operations, through `bitsandbytes.nn.Linear8bitLt` and `bitsandbytes.nn.Linear4bit` and 8-bit optimizers through `bitsandbytes.optim` module.
88

9-
This fork is actively developed for ROCm and updates are being pushed into `multi-backend-refactor` branch of upstream bitsandbytes. Users can use either of these to run bitsandbytes on AMD GPUs.
9+
This fork is developed for ROCm and updates are pushed into upstream. Users can use either of these to run bitsandbytes on AMD GPUs.
1010

11-
**Note: The default branch of this fork is switched from `rocm_enabled` to `rocm_enabled_multi_backend`. This is synced periodically with `multi-backend-refactor` branch of upstream, and latest developments are pushed here until upstream branch is merged into `main`.**
11+
**Note: The default branch of this fork is switched from this branch to the new `rocm_enabled` branch. **
1212

1313
**Installation for ROCm:**
1414

1515
For latest develop version:
1616
```bash
1717
git clone --recurse https://github.com/ROCm/bitsandbytes
1818
cd bitsandbytes
19-
git checkout rocm_enabled_multi_backend
20-
pip install -r requirements-dev.txt
19+
git checkout rocm_enabled
2120
cmake -DCOMPUTE_BACKEND=hip -S . #Use -DBNB_ROCM_ARCH="gfx90a;gfx942" to target specific gpu arch
2221
make
2322
pip install .

0 commit comments

Comments
 (0)