bitsandbytes
bitsandbytes copied to clipboard
[BUG] AMD GPU compilation from source
System Info
Linux Mint 21.3 x86_64 Python 3.10.12 AMD Ryzen 5 1600 AMD Radeon RX 7900XTX
Reproduction
Run the commands in the section: Compiling from Source (AMD GPU).
Using the AMD 7900XTX, run cmake -DBNB_ROCM_ARCH="gfx942" -DCOMPUTE_BACKEND=hip -S ..
The following error message will be returned:
cmake -DBNB_ROCM_ARCH="gfx942" -DCOMPUTE_BACKEND=hip -S . # Use -DBNB_ROCM_ARCH="gfx90a;gfx942" to target specific gpu arch
-- Configuring bitsandbytes (Backend: hip)
-- NO_CUBLASLT := OFF
-- The HIP compiler identification is unknown
CMake Error at /usr/local/share/cmake-3.31/Modules/CMakeDetermineHIPCompiler.cmake:174 (message):
Failed to find ROCm root directory.
Call Stack (most recent call first):
CMakeLists.txt:193 (enable_language)
Expected behavior
The compilation should finish without an error message.