summaryrefslogtreecommitdiff
path: root/modules/nixos/mixins/nvidia.nix
blob: ff813859b4364611259c05678e82d9e724d9d698 (plain)
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
{ config, lib, ... }:

let
  cfg = config.hardware.nvidia;

  isNvidiaEnabled = lib.elem "nvidia" config.services.xserver.videoDrivers;

  # Unlike Nixpkgs, I know all of my GPUs should prefer the open modules after 560
  useOpenModulesByDefault = lib.versionAtLeast config.hardware.nvidia.package.version "560";
in

{
  options.hardware.nvidia = {
    nvk.enable = lib.mkEnableOption "an NVK specialisation";
  };

  config = lib.mkMerge [
    {
      hardware.nvidia = {
        open = useOpenModulesByDefault;
        # We usually want this to make suspend, etc. work
        powerManagement.enable = lib.mkDefault true;
      };
    }

    (lib.mkIf (isNvidiaEnabled && !config.hardware.nvidia.open) {
      # Don't use GSP Firmware on proprietary driver
      # https://github.com/NVIDIA/open-gpu-kernel-modules/issues/693
      boot.kernelParams = [
        "nvidia.NVreg_EnableGpuFirmware=0"
      ];
    })

    (lib.mkIf cfg.nvk.enable {
      specialisation = {
        nvk.configuration = {
          boot = {
            # required for GSP firmware
            kernelParams = [ "nouveau.config=NvGspRm=1" ];
            # we want early KMS
            # https://wiki.archlinux.org/title/Kernel_mode_setting#Early_KMS_start
            initrd.kernelModules = [ "nouveau" ];
          };

          hardware = {
            graphics.extraPackages = lib.mkForce [ ];
            nvidia-container-toolkit.enable = lib.mkForce false;
          };

          services.xserver.videoDrivers = lib.mkForce [ "modesetting" ];

          system.nixos.tags = [ "with-nvk" ];
        };
      };
    })

    (lib.mkIf config.traits.containers.enable {
      hardware.nvidia-container-toolkit.enable = true;
    })
  ];
}