nixos-lib/hardware/nvidia/proprietary.nix

{ config, pkgs, lib, ... }:
with lib;
let
  cfg = config.aviallon.hardware.nvidia;
in {
    config = mkIf (cfg.enable && cfg.useProprietary) {
      boot.initrd.kernelModules = [
        "nvidia"
        "nvidia_drm"
        "nvidia_uvm"
        "nvidia_modeset"
      ];
      # boot.blacklistedKernelModules = [ "nouveau" ];
      services.xserver.videoDrivers = [
        "nvidia"
      ];

      hardware.nvidia = {
        powerManagement.enable = true;
        powerManagement.finegrained = mkIf config.hardware.nvidia.prime.offload.enable true;
        modesetting.enable = true;
      };

      aviallon.boot.cmdline = mkIf cfg.saveAllVram {
        NVreg_PreserveVideoMemoryAllocations = 1;
        NVreg_TemporaryFilePath = "/tmp/nvidia-gpu.vram.img";
      };

      aviallon.programs.allowUnfreeList = [
          "nvidia-x11"
          "nvidia-settings"
          "cudatoolkit"
      ];

      hardware.opengl.extraPackages = with pkgs; [
        libvdpau-va-gl
        vaapiVdpau
      ];

      hardware.opengl.extraPackages32 = with pkgs.pkgsi686Linux; [
        libvdpau-va-gl
        vaapiVdpau
      ];

      environment.systemPackages = with pkgs; [
        nvtop
      ];

      # See documentation here: https://download.nvidia.com/XFree86/Linux-x86_64/510.60.02/README/openglenvvariables.html
      environment.variables = {
        "__GL_YIELD" = "USLEEP"; # use usleep(0) instead of sched_yield() -> better performance in most cases
        "__GL_ALLOW_UNOFFICIAL_PROTOCOL" = "1"; # allow unofficial GLX protocol if also set in Xorg conf
        "__GL_VRR_ALLOWED" = "1"; # Try to enable G-SYNC VRR if screen AND app is compatible
        "KWIN_DRM_USE_EGL_STREAMS"  = "1"; # Make KWin use EGL Streams, because otherwise performance will be horrible.
      };
    };
}