gador/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ollama-rocm: fix evaluation error when rocmPackages are built for a specific arch (#405457)

Browse Source
This commit is contained in:
Pavol Rusnak
2025-05-09 19:14:33 +02:00
committed by GitHub
parent 5309e4e870 ac5c65a71a
commit eacfb13d85
1 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/by-name/ol/ollama/package.nix
+1 -7
View File
@@ -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 =