ollama: test that rocm and cuda build successfully

This commit is contained in:
abysssol 2024-02-25 13:22:28 -05:00
parent b8d8c1f207
commit 59a240579c

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";