nixos/tenstorrent: init

nixos/modules/hardware/tenstorrent.nix

@@ -0,0 +1,29 @@
+{
+  config,
+  pkgs,
+  lib,
+  ...
+}:
+let
+  inherit (lib) mkEnableOption mkIf;
+  inherit (config.boot.kernelPackages) tt-kmd;
+
+  cfg = config.hardware.tenstorrent;
+in
+{
+  options.hardware.tenstorrent.enable = mkEnableOption "Tenstorrent driver & utilities";
+
+  config = mkIf cfg.enable {
+    boot = {
+      extraModulePackages = [ tt-kmd ];
+      kernelModules = [ "tenstorrent" ];
+    };
+
+    services.udev.packages = [
+      tt-kmd
+    ];
+
+    # TODO: add tt-smi to environment.systemPackages once https://github.com/NixOS/nixpkgs/pull/444714 is merged
+    # TODO: add tt-system-tools to environment.systemPackages once https://github.com/NixOS/nixpkgs/pull/444748 is merged
+  };
+}

nixos/modules/module-list.nix

@@ -109,6 +109,7 @@
   ./hardware/sheep-net.nix
   ./hardware/steam-hardware.nix
   ./hardware/system-76.nix
+  ./hardware/tenstorrent.nix
   ./hardware/tuxedo-drivers.nix
   ./hardware/ubertooth.nix
   ./hardware/uinput.nix