Merge pull request #323900 from abysssol/ollama-rocm-libpath

ollama: add `rocmPath` to `LD_LIBRARY_PATH` in wrapper
This commit is contained in:
Pol Dellaiera 2024-07-02 05:42:03 +02:00 committed by GitHub
commit d70aec8605
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -126,6 +126,7 @@ let
# until these llama-cpp binaries can have their runpath patched
"--suffix LD_LIBRARY_PATH : '${addDriverRunpath.driverLink}/lib'"
] ++ lib.optionals enableRocm [
"--suffix LD_LIBRARY_PATH : '${rocmPath}'"
"--set-default HIP_PATH '${rocmPath}'"
];
wrapperArgs = builtins.concatStringsSep " " wrapperOptions;