1 files changed, 17 insertions, 0 deletions
diff --git a/nixos/tests/ollama-rocm.nix b/nixos/tests/ollama-rocm.nix
new file mode 100644
index 0000000000000..81915630d950b
--- /dev/null
+++ b/nixos/tests/ollama-rocm.nix
@@ -0,0 +1,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)
+  '';
+}