From 34e8b3aae648779c0149af7632ba2913603b84b2 Mon Sep 17 00:00:00 2001
From: Artturin <Artturin@artturin.com>
Date: Mon, 14 Oct 2024 03:31:44 +0300
Subject: [PATCH] Add `nixosModules.tt-module`

---
 flake.nix | 28 +++++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/flake.nix b/flake.nix
index 26d01b0..a229724 100644
--- a/flake.nix
+++ b/flake.nix
@@ -80,5 +80,31 @@
 
         formatter = pkgs.nixfmt-rfc-style;
       }
-    );
+    )
+    // {
+      # TODO: Add overlay
+      nixosModules = {
+        tt-module =
+          { config, pkgs, ... }:
+          {
+            boot = {
+              extraModulePackages = [
+                ((pkgs.tt-kmd or self.packages.${pkgs.hostPlatform.system}.kmd).override {
+                  kernel = config.boot.kernelPackages.kernel;
+                })
+              ];
+              kernelParams = [
+                "hugepagesz=1G"
+                "hugepages=2"
+                "iommu=pt"
+              ];
+              kernelModules = [ "tenstorrent" ];
+            };
+            services.udev.packages = [
+              (pkgs.tt-udev-rules or self.packages.${pkgs.hostPlatform.system}.udev-rules)
+            ];
+
+          };
+      };
+    };
 }