Skip to content

Commit 068b1ee

Browse files
authored
ollama: use cudaPackages.backendStdenv (NixOS#375388)
2 parents 2fc5aeb + ad5309a commit 068b1ee

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

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

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,10 @@ let
129129
wrapperArgs = builtins.concatStringsSep " " wrapperOptions;
130130

131131
goBuild =
132-
if enableCuda then buildGoModule.override { stdenv = overrideCC stdenv gcc12; } else buildGoModule;
132+
if enableCuda then
133+
buildGoModule.override { stdenv = cudaPackages.backendStdenv; }
134+
else
135+
buildGoModule;
133136
inherit (lib) licenses platforms maintainers;
134137
in
135138
goBuild {
@@ -146,9 +149,7 @@ goBuild {
146149
CLBlast_DIR = "${clblast}/lib/cmake/CLBlast";
147150
HIP_PATH = rocmPath;
148151
}
149-
// lib.optionalAttrs enableCuda {
150-
CUDA_PATH = cudaPath;
151-
};
152+
// lib.optionalAttrs enableCuda { CUDA_PATH = cudaPath; };
152153

153154
nativeBuildInputs =
154155
[
@@ -225,9 +226,7 @@ goBuild {
225226

226227
__darwinAllowLocalNetworking = true;
227228

228-
nativeInstallCheck = [
229-
versionCheckHook
230-
];
229+
nativeInstallCheck = [ versionCheckHook ];
231230
versionCheckProgramArg = [ "--version" ];
232231
doInstallCheck = true;
233232

0 commit comments

Comments
 (0)