Skip to content

Commit dfbd681

Browse files
authored
Install bitsandbytes for ROCm (#1915)
Update ROCm regression test workflow to uninstall and force-reinstall bitsandbytes for rocm
1 parent 40a6867 commit dfbd681

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

.github/workflows/regression_test_rocm.yml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,8 @@ jobs:
4343
python -m pip install --upgrade pip
4444
pip install ${{ matrix.torch-spec }}
4545
pip install -r dev-requirements.txt
46+
pip uninstall -y bitsandbytes
47+
pip install --force-reinstall 'https://github.com/bitsandbytes-foundation/bitsandbytes/releases/download/continuous-release_multi-backend-refactor/bitsandbytes-0.44.1.dev0-py3-none-manylinux_2_24_x86_64.whl'
4648
pip install .
4749
export CONDA=$(dirname $(dirname $(which conda)))
4850
export LD_LIBRARY_PATH=$CONDA/lib/:$LD_LIBRARY_PATH

0 commit comments

Comments
 (0)