dotfiles

My PC Dotfiles & Website
git clone https://github.com/maydayv7/dotfiles

gpu.nix (2586B)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
## GPU Configuration ##
_: {
  flake.modules.nixos.gpu = {
    config,
    lib,
    pkgs,
    ...
  }: let
    inherit
      (lib)
      mkForce
      mkIf
      mkEnableOption
      mkMerge
      mkOption
      types
      ;
    cfg = config.hardware.gpu;
    mode = config.hardware.cpu.mode != "powersave";
    gamemode = config.programs.gamemode.enable;
  in {
    options.hardware.gpu = {
      enable = mkEnableOption "Discrete GPU Support";
      model = mkOption {
        description = "Discrete GPU Model";
        type = with types; nullOr (enum ["nvidia"]);
        default = null;
      };
    };

    config = mkMerge [
      (mkIf cfg.enable {
        assertions = [
          {
            assertion = cfg.model != null;
            message = "GPU model must be set when GPU is enabled";
          }
        ];
      })

      (mkIf (cfg.enable && cfg.model == "nvidia") (
        let
          hybrid = with config.hardware.nvidia.prime; (amdgpuBusId != "" || intelBusId != "");
        in {
          services.xserver.videoDrivers = mkForce ["nvidia"];
          environment = {
            systemPackages = [pkgs.btop-cuda];
            variables."__GLX_VENDOR_LIBRARY_NAME" = mkIf mode "nvidia";
            sessionVariables."GAMEMODERUNEXEC" = mkIf (hybrid && gamemode && !mode) "nvidia-offload";
          };

          boot = {
            blacklistedKernelModules = ["nouveau"];
            kernelModules = [
              "nvidia"
              "nvidia_drm"
              "nvidia_modeset"
              "nvidia_uvm"
            ];
            kernelParams = ["nvidia-drm.fbdev=1"];
            extraModprobeConfig = ''
              # PAT Support
              options nvidia NVreg_UsePageAttributeTable=1

              # DDC/CI Support
              options nvidia NVreg_RegistryDwords=RMUseSwI2c=0x01;RMI2cSpeed=100
            '';
          };

          # Wayland Support
          hardware.nvidia = {
            modesetting.enable = mkForce true;
            nvidiaSettings = mkForce true;
            dynamicBoost.enable = true;
            powerManagement.enable = true;
            prime = mkIf hybrid {
              sync.enable = mkForce mode;
              offload = {
                enable = mkForce (!mode);
                enableOffloadCmd = mkForce (!mode);
              };
            };
          };
        }
      ))

      (mkIf (!cfg.enable && cfg.model == "nvidia") {
        boot.blacklistedKernelModules = [
          "nvidia"
          "nvidia_drm"
          "nvidia_modeset"
          "nvidia_uvm"
        ];
      })
    ];
  };
}