Unverified Commit efbd3247 authored by Pol Dellaiera's avatar Pol Dellaiera Committed by GitHub
Browse files

Merge pull request #298077 from ulrikstrid/ulrikstrid--rocm-6-ollama-2

ollama: Fix rocm runtime issues
parents 875d4716 99ca7e3a
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
@@ -165,7 +165,8 @@ goBuild ((lib.optionalAttrs enableRocm {
    # expose runtime libraries necessary to use the gpu
    mv "$out/bin/ollama" "$out/bin/.ollama-unwrapped"
    makeWrapper "$out/bin/.ollama-unwrapped" "$out/bin/ollama" \
      --suffix LD_LIBRARY_PATH : '/run/opengl-driver/lib:${lib.makeLibraryPath runtimeLibs}'
      --suffix LD_LIBRARY_PATH : '/run/opengl-driver/lib:${lib.makeLibraryPath runtimeLibs}' '' + lib.optionalString enableRocm ''\
      --set-default HIP_PATH ${pkgs.rocmPackages.meta.rocm-hip-libraries}
  '';

  ldflags = [