1 files changed, 61 insertions, 0 deletions
diff --git a/modules/hardware/gpu.nix b/modules/hardware/gpu.nix
new file mode 100644
index 0000000..d7dfe17
--- /dev/null
+++ b/modules/hardware/gpu.nix
@@ -0,0 +1,61 @@
+{
+    config,
+    options,
+    lib,
+    pkgs,
+    ...
+}: let
+    gpuConfig = config.modules.hardware.gpu;
+    device = config.modules.device;
+in {
+    options.modules.hardware.gpu = {
+        enable = lib.mkOption {
+            type = lib.types.bool;
+            default = true;
+        };
+    };
+
+    config = lib.mkIf (gpuConfig.enable) (lib.mkMerge [
+        {
+            hardware.opengl = {
+                enable = true;
+                driSupport = true;
+                driSupport32Bit = true;
+            };
+        }
+        (lib.mkIf (device.gpu == "intel") {
+            boot.initrd.kernelModules = ["i915"];
+            services.xserver.videoDrivers = ["modesetting"];
+
+            hardware.opengl.extraPackages = [
+                pkgs.intel-compute-runtime
+                pkgs.intel-media-driver
+                pkgs.vaapiIntel
+                pkgs.vaapiVdpau
+                pkgs.libvdpau-va-gl
+            ];
+
+            environment.variables.VDPAU_DRIVER = "va_gl";
+        })
+        (lib.mkIf (device.gpu == "nvidia") {
+            services.xserver.videoDrivers = ["nvidia"];
+            hardware.nvidia = {
+                modesetting.enable = true;
+                powerManagement.enable = false;
+                powerManagement.finegrained = false;
+                open = false;
+                nvidiaSettings = true;
+                package = config.boot.kernelPackages.nvidiaPackages.beta;
+            };
+            boot.initrd.kernelModules = [
+                "nvidia"
+                "nvidia_modeset"
+                "nvidia_uvm"
+                "nvidia_drm"
+            ];
+            boot.extraModprobeConfig = ''
+                options modeset=1 fbdev=1
+            '';
+        })
+    ]);
+}