nixos/tests/ollama-rocm.nix

   1 { lib, ... }:
   2 {
   3   name = "ollama-rocm";
   4   meta.maintainers = with lib.maintainers; [ abysssol ];
   5
   6   nodes.rocm =
   7     { ... }:
   8     {
   9       services.ollama.enable = true;
  10       services.ollama.acceleration = "rocm";
  11     };
  12
  13   testScript = ''
  14     rocm.wait_for_unit("multi-user.target")
  15     rocm.wait_for_open_port(11434)
  16   '';
  17 }