mirrors/nixpkgs
1
0
Fork 1
mirror of https://github.com/NixOS/nixpkgs.git synced 2024-09-11 15:08:33 +01:00
Code Issues Wiki Activity

ollama: test that rocm and cuda build successfully

Browse source
This commit is contained in:
abysssol 2024-02-25 13:22:28 -05:00
parent b8d8c1f207
commit 59a240579c
1 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
pkgs/tools/misc/ollama/default.nix
View file

@ -8,6 +8,7 @@
, makeWrapper
, stdenv
, pkgs
, cmake
, gcc12
, clblast
@ -163,6 +164,12 @@ goBuild ((lib.optionalAttrs enableRocm {
"-X=github.com/jmorganca/ollama/server.mode=release"
];
# for now, just test that rocm and cuda build
passthru.tests = lib.optionalAttrs stdenv.isLinux {
rocm = pkgs.ollama.override { acceleration = "rocm"; };
cuda = pkgs.ollama.override { acceleration = "cuda"; };
};
meta = {
description = "Get up and running with large language models locally";
homepage = "https://github.com/jmorganca/ollama";