ollama: add rocmPath to LD_LIBRARY_PATH in wrapper

Pr #323249 removed `rocmPath` from `LD_LIBRARY_PATH`, but the comment
> since `rocmPath` is in `LD_LIBRARY_PATH`, ollama uses rocm correctly

makes me think I already tested this and that it doesn't work
This commit is contained in:
abysssol
2024-07-01 19:24:35 -04:00
parent f56b0eb2fc
commit 559cffd864
+1
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;