Commit 47348f55 authored by Tristan Ross's avatar Tristan Ross
Browse files

nixos/tenstorrent: init

parent 70d95f98

nixos/modules/hardware/tenstorrent.nix

0 → 100644
+29 −0
Original line number Diff line number Diff line 
{

  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

+1 −0
Original line number Diff line number Diff line
@@ -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