From 7e5aefead8451ac50d74ed9b5e3a7953b6f4f193 Mon Sep 17 00:00:00 2001
From: shynd <shynd@r-io.lu>
Date: Thu, 20 Feb 2025 18:23:06 +0100
Subject: [PATCH] modules - nvidia prime drivers

---
 modules/nvidia-prime-drivers.nix | 33 ++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 modules/nvidia-prime-drivers.nix

diff --git a/modules/nvidia-prime-drivers.nix b/modules/nvidia-prime-drivers.nix
new file mode 100644
index 0000000..77ed0ac
--- /dev/null
+++ b/modules/nvidia-prime-drivers.nix
@@ -0,0 +1,33 @@
+{ lib, pkgs, config, ... }:
+
+with lib;
+let
+  cfg = config.drivers.nvidia-prime;
+in
+{
+  options.drivers.nvidia-prime = {
+    enable = mkEnableOption "enable nvidia prime hybrid gpu offload";
+    intelBusID = mkOption {
+      type = types.str;
+      default = "PCI:1:0:0";
+    };
+    nvidiaBusID = mkOption {
+      type = types.str;
+      default = "PCI:0:2:0";
+    };
+  };
+
+  config = mkIf cfg.enable {
+    hardware.nvidia = {
+      prime = {
+        offload = {
+          enable = true;
+          enableOffloadCmd = true;
+        };
+        # make sure to use the correct BusID values for your system!!!
+        intelBusID = "${cfg.intelBusID}";
+        nvidiaBusID = "${cfg.nvidiaBusID}";
+      };
+    };
+  };
+}
\ No newline at end of file