2c2c36d3da
Adds a new toolbox variant based on PR #21344 (pedapudi/llama.cpp@gfx1151-opt) which tunes MMQ tile sizes (x_max=48, y=64) and warp counts (nwarps=4) for RDNA3_5 gfx1151, yielding up to +100% prefill throughput at small batch sizes. Also adds BMI2/FMA/F16C CPU SIMD flags and GGML_CUDA_FA_ALL_QUANTS=ON to match the benchmark build used in the PR. Wire up CI (build matrix + prune), the refresh script, and run_benchmarks.sh so results land alongside rocm-7.2.1.
2.8 KiB
2.8 KiB
AI Agent Context: AMD Strix Halo Llama.cpp Toolboxes
Primary Goal: This project provides pre-built containers ("toolboxes") for running llama.cpp optimally on AMD Ryzen AI Max "Strix Halo" APUs. It simplifies environment setup for ROCm and Vulkan backends, allowing users to leverage up to 124 GiB of unified system memory for LLM inference.
Core Technologies
- Containerization: Toolbx (Fedora) or Distrobox (Ubuntu). Underneath, Docker/Podman is used to build base images.
- Inference Engine: Llama.cpp
- Hardware / Drivers: AMD "Strix Halo" APUs (Gfx1151). Implementations depend on ROCm (v6.4.4, v7.x) and Vulkan (Mesa RADV, AMDVLK).
Repository Structure Overview
/toolboxes/: Dockerfiles used to build the container images (e.g.,rocm-6.4.4,rocm-7.2.1,vulkan-radv). These often use multi-stage builds to compile Llama.cpp and extract standalone binaries./benchmark/: Shell scripts and Python utilities (likegenerate_results_json.py) to systematically test Llama.cpp throughput, latency, and RPC performance./docs/: Markdown documents, along with HTML/CSS/JS (e.g.,index.html,assets/) for the GitHub Pages website (strix-halo-toolboxes.com), plus interactive benchmark viewers and documentation on VRAM estimation./scripts/: Python utilities, includingrun_distributed_llama.pyfor distributed inference across nodes..github/workflows/: GitHub Actions that automatically rebuild containers whenever the upstreamllama.cppmaster branch updates or when triggered manually.
Critical Technical Quirks (Important for Development)
- Flash Attention & no-mmap: Running
llama-serverorllama-clion Strix Halo requires-fa 1(flash attention) and--no-mmapto avoid memory fragmentation and crashes. - Kernel memory params: The optimal Strix Halo host configuration relies on custom boot parameters (
iommu=pt amdgpu.gttsize=126976 ttm.pages_limit=32505856) to allocate unified RAM to the iGPU. - ROCm 7+ Workaround: Due to LLVM compiler regressions, ROCm 7 builds currently use a workaround flag (
-mllvm --amdgpu-unroll-threshold-local=600) to restore Llama.cpp performance. - Kernel Bugs: Avoid kernels older than 6.18.4, and the specifically broken
linux-firmware-20251125.
General Instructions for Coding Agents
- Container Builds: When modifying
Dockerfile.*files inside/toolboxes, ensure the build output remains lean and only necessary runtime dependencies and Llama.cpp binaries are carried over. - Documentation Synchronization: If adding a new backend or feature, ensure
README.mdis updated simultaneously. - Scripts: Benchmarking and utility scripts are expected to integrate with standard
toolboxexecution. Use/dev/driand/dev/kfdmounts for device access.