nix: Added missing packages and options for ROCm build #15747
+28
−8
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Currently, Nix flake for llama-cpp-rocm doesn't build due to missing
lld
andobjcopy
executables. This MR fixes it by adding them to package list.Additionally, this MR:
mpi
package depending on selected backendbuilsAllCudaFaQuants
parameter to enable building all flash attention quants for CUDA and ROCm builds.rocmUseWmma
to add ROCWMMA_FATTN support, enabled by default.enableUma
parameter to enable UMA support for CUDA/ROCm platformsIn order to enable WMMA support, i had to add a CMake variable and tweak the build script to provide a way of setting the path to include directory manually - otherwise CMake could not find the headers and failed the build
I've tested building the flake with/without ROCm and MPI support and it works fine now and produces working llama-server.