ollama: test that rocm and cuda build successfully
This commit is contained in:
parent
b8d8c1f207
commit
59a240579c
1 changed files with 7 additions and 0 deletions
|
@ -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";
|
||||
|
|
Loading…
Reference in a new issue