Changed layouts

modules/hardware/nvidia.nix

@@ -1,26 +1,42 @@
-{ config, ... }:
+{ config, lib, pkgs, ... }:
 
-{
-  hardware.graphics = { enable = true; };
-
-  services.xserver.videoDrivers = [ "nvidia" ];
-
-  hardware.nvidia = {
-    modesetting.enable = true;
-    powerManagement.enable = false;
-    open = false;
-    nvidiaSettings = true;
-    package = config.boot.kernelPackages.nvidiaPackages.stable;
+let inherit (lib) mkOption types;
+in {
+  options = {
+    enableCuda = mkOption {
+      type = types.bool;
+      default = false;
+      description = ''
+        Enable CUDA support for Blender and other packages.
+        Warning: This can cause very long build times.
+      '';
+    };
+    nvidiaOpen = mkOption {
+      type = types.bool;
+      default = false;
+    };
   };
 
-  # Cuda for blender long build times
-  #environment.systemPackages = with pkgs; [
-  #	(blender.override {
-  #		cudaSupport = true;
-  #	})
-  #];
+  config = {
+    hardware.graphics.enable = true;
 
-  # This causes very long build times
-  #nixpkgs.config.cudaSupport = true;
-  #hardware.enableAllFirmware = true;
+    services.xserver.videoDrivers = [ "nvidia" ];
+
+    hardware.nvidia = {
+      modesetting.enable = true;
+      powerManagement.enable = false;
+      open = config.nvidiaOpen;
+      nvidiaSettings = true;
+      package = config.boot.kernelPackages.nvidiaPackages.stable;
+    };
+
+    # Apply CUDA-specific settings if enabled
+    nixpkgs.config.cudaSupport = config.enableCuda;
+    hardware.enableAllFirmware = config.enableCuda;
+
+    environment.systemPackages = (if config.enableCuda then
+      [ (pkgs.blender.override { cudaSupport = true; }) ]
+    else
+      [ ]);
+  };
 }