cleanup docs-build breaking install instructs

Titus-von-Koeller · Jun 6, 2024 · a51863e · a51863e
1 parent 36fe1a0
commit a51863e
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 46 deletions.
diff --git a/docs/source/installation.mdx b/docs/source/installation.mdx
@@ -91,6 +91,32 @@ Big thanks to [wkpark](https://github.com/wkpark), [Jamezo97](https://github.com
 </hfoption>
 </hfoptions>
 
+## Multi-backend preview release (+ compilation)
+
+Please follow these steps to install bitsandbytes with device-specific backend support other than CUDA:
+
+<hfoptions id="Alternative backend">
+<hfoption id="AMD ROCm">
+
+For a ROCm specific install:
+
+```bash
+git clone --depth 1 https://github.com/TimDettmers/bitsandbytes.git && cd bitsandbytes/
+git checkout multi-backend-refactor
+
+# Install dependencies
+apt install hipblaslt
+pip install -r requirements-dev.txt
+
+# Compile & install
+cmake -DCOMPUTE_BACKEND=hip -S . #Use -DBNB_ROCM_ARCH="gfx90a;gfx942" to target specific gpu arch
+make
+pip install .
+```
+
+<hfoption>
+</hfoptions>
+
 ## PyTorch CUDA versions
 
 Some bitsandbytes features may need a newer CUDA version than the one currently supported by PyTorch binaries from Conda and pip. In this case, you should follow these instructions to load a precompiled bitsandbytes binary.

diff --git a/docs/source/rocm_installation.mdx b/docs/source/rocm_installation.mdx