nixos/tests/ollama-rocm.nix


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

{ lib, ... }:
{
  name = "ollama-rocm";
  meta.maintainers = with lib.maintainers; [ abysssol ];

  nodes.rocm =
    { ... }:
    {
      services.ollama.enable = true;
      services.ollama.acceleration = "rocm";
    };

  testScript = ''
    rocm.wait_for_unit("multi-user.target")
    rocm.wait_for_open_port(11434)
  '';
}