c55aaa5809
BNB_ROCM_VERSION was pinned to '72' which works today (AMD wheel ships
rocm72.dll) but would break again if AMD ships a future wheel with a
different DLL suffix (e.g. rocm713.dll).
Add _detect_bnb_rocm_dll_ver() to install_python_stack.py: scans the
installed bitsandbytes package dir for libbitsandbytes_rocm{VER}.dll
using importlib.util.find_spec (no BNB import needed) and returns the
suffix. '72' remains the fallback when detection fails.
Apply the same detection inline in worker.py section 1f. Both paths
still respect a pre-set BNB_ROCM_VERSION (caller override wins).
Tests: +8 cases covering detection logic and fallback (147 passed, 2 skipped).
|
||
|---|---|---|
| .. | ||
| conftest.py | ||
| smoke_test_llama_prebuilt.py | ||
| smoke_test_parallel_studio_home.py | ||
| test_install_llama_prebuilt_logic.py | ||
| test_llama_pr_force_and_source.py | ||
| test_pr4562_bugfixes.py | ||
| test_rocm_support.py | ||
| test_selection_logic.py | ||