Graduate eirene

Signed-off-by: Nikolaos Karaolidis <nick@karaolidis.com>

hosts/jupiter/hardware/default.nix

@@ -1,18 +1,75 @@
-{ ... }:
 {
-  # TODO Remove
-  hardware.enableAllHardware = true;
+  config,
+  pkgs,
+  lib,
+  ...
+}:
+{
+  imports = [ ./display.nix ];
 
-  # TODO Update
-  boot.initrd.kernelModules = [
-    "ahci"
-    "xhci_pci"
-    "virtio_pci"
-    "virtio_scsi"
-    "virtio_blk"
-    "virtio_console"
-    "sd_mod"
-  ];
+  hardware = {
+    enableAllFirmware = true;
 
-  services.fstrim.enable = true;
+    cpu = {
+      cores = 8;
+      threads = 16;
+      amd.updateMicrocode = true;
+    };
+
+    nvidia = {
+      open = true;
+      powerManagement.enable = true;
+      dynamicBoost.enable = true;
+
+      prime = {
+        offload = {
+          enable = true;
+          enableOffloadCmd = true;
+        };
+        nvidiaBusId = "PCI:1:0:0";
+        amdgpuBusId = "PCI:6:0:0";
+      };
+    };
+
+    graphics = {
+      enable32Bit = true;
+      extraPackages = with pkgs; [
+        amdvlk
+        driversi686Linux.amdvlk
+        rocmPackages.clr
+        rocmPackages.clr.icd
+      ];
+    };
+
+    nvidia-container-toolkit.enable = (
+      config.virtualisation.containerd.enable
+      || config.virtualisation.docker.enable
+      || config.virtualisation.podman.enable
+    );
+  };
+
+  boot = {
+    kernelParams = [ "amd_pstate=active" ];
+    kernelModules = [ "kvm-amd" ];
+    initrd.kernelModules = [
+      "nvme"
+      "ahci"
+      "usbhid"
+      "amdgpu"
+    ];
+  };
+
+  nixpkgs.config = {
+    cudaSupport = true;
+    rocmSupport = true;
+  };
+
+  powerManagement.enable = true;
+
+  services = {
+    xserver.videoDrivers = [ "nvidia" ];
+    fstrim.enable = true;
+    tlp.settings.DISK_DEVICES = lib.mkDefault "nvme0n1 nvme1n1";
+    logind.lidSwitch = "ignore";
+  };
 }