Unverified Commit ac5c65a7 authored by Luna Nova's avatar Luna Nova
Browse files

ollama-rocm: fix evaluation error when clr.localGpuTargets is set

parent 62cab5db

pkgs/by-name/ol/ollama/package.nix

+1 −7
Original line number Diff line number Diff line
@@ -14,7 +14,7 @@ 
  clblast,

  libdrm,

  rocmPackages,

  rocmGpuTargets ? rocmPackages.clr.gpuTargets or [ ],

  rocmGpuTargets ? rocmPackages.clr.localGpuTargets or (rocmPackages.clr.gpuTargets or [ ]),

  cudaPackages,

  cudaArches ? cudaPackages.cudaFlags.realArches or [ ],

  autoAddDriverRunpath,
@@ -137,12 +137,6 @@ goBuild (finalAttrs: { 
      CFLAGS = "-Wno-c++17-extensions -I${rocmPath}/include";

      CXXFLAGS = "-Wno-c++17-extensions -I${rocmPath}/include";

    }

    // lib.optionalAttrs (enableRocm && (rocmPackages.clr.localGpuTargets or false)) {



      # If rocm CLR is set to build for an exact set of targets reuse that target list,

      # otherwise let ollama use its builtin defaults

      HIP_ARCHS = lib.concatStringsSep ";" rocmPackages.clr.localGpuTargets;

    }

    // lib.optionalAttrs enableCuda { CUDA_PATH = cudaPath; };



  nativeBuildInputs =