86b1918056setup.sh and setup.ps1 resolve the AMD gfx target (rocminfo/hipinfo/amd-smi, name inference, or UNSLOTH_ROCM_GFX_ARCH) but never passed it to install_llama_prebuilt.py. The installer re-probed on its own and, on hosts where hipinfo/amd-smi cannot report the arch (amd-smi-only Linux, HIP-runtime-only Strix Halo, name-inferred GPUs), left rocm_gfx_target unset. The lemonade HIP prebuilt selection needs that arch, so those hosts got no GPU prebuilt and fell back to a source build. Add a --rocm-gfx argument (defaulting to UNSLOTH_ROCM_GFX_ARCH) and fold it into the host profile via _apply_host_overrides: a forwarded gfx is authoritative (setup already applied visible-device selection) and implies has_rocm. setup.sh and setup.ps1 now forward their resolved arch. Add unit tests for _normalize_forwarded_gfx and _apply_host_overrides, plus source checks that both setup scripts forward --rocm-gfx. |
||
|---|---|---|
| .. | ||
| backend | ||
| frontend | ||
| src-tauri | ||
| __init__.py | ||
| install_llama_prebuilt.py | ||
| install_python_stack.py | ||
| LICENSE.AGPL-3.0 | ||
| package-lock.json | ||
| package.json | ||
| setup.bat | ||
| setup.ps1 | ||
| setup.sh | ||
| Unsloth_Studio_Colab.ipynb | ||