ollama: add nixos test to passthru.tests

This commit is contained in:
abysssol 2024-03-11 21:01:47 -04:00
parent efed30f903
commit af67cfa87b

View File

@ -7,6 +7,7 @@
, overrideCC
, makeWrapper
, stdenv
, nixosTests
, pkgs
, cmake
@ -163,8 +164,8 @@ 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 {
passthru.tests = {
service = nixosTests.ollama;
rocm = pkgs.ollama.override { acceleration = "rocm"; };
cuda = pkgs.ollama.override { acceleration = "cuda"; };
};