Commit ea8ab5ba authored by abysssol's avatar abysssol
Browse files

ollama: fix a symlink and use a smaller rocm path

parent ddd523bd

pkgs/tools/misc/ollama/default.nix

+5 −1
Original line number Diff line number Diff line
@@ -68,6 +68,10 @@ let 
      cudaPackages.cudatoolkit

      cudaPackages.cuda_cudart

    ];

    postBuild = ''

      rm "$out/lib64"

      ln -s "lib" "$out/lib64"

    '';

  };



  runtimeLibs = lib.optionals enableRocm [
@@ -166,7 +170,7 @@ goBuild ((lib.optionalAttrs enableRocm { 
    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}' '' + lib.optionalString enableRocm ''\

      --set-default HIP_PATH ${pkgs.rocmPackages.meta.rocm-hip-libraries}

      --set-default HIP_PATH ${rocmPath}

  '';



  ldflags = [