147 lines
3.7 KiB
Nix
147 lines
3.7 KiB
Nix
{
|
|
config,
|
|
lib,
|
|
pkgs,
|
|
...
|
|
}: let
|
|
inherit (lib) mkEnableOption mkOption types mkIf mkMerge flatten concatMap;
|
|
|
|
cfg = config.nixos.hardware.graphics;
|
|
|
|
commonPackages = with pkgs; [
|
|
libva
|
|
vaapiVdpau
|
|
libvdpau-va-gl
|
|
libGL
|
|
];
|
|
|
|
commonPackages32 = with pkgs.pkgsi686Linux; [
|
|
libva
|
|
vaapiVdpau
|
|
libvdpau-va-gl
|
|
];
|
|
|
|
mesaVulkanPackages = with pkgs; [
|
|
vulkan-loader
|
|
vulkan-validation-layers
|
|
vulkan-extension-layer
|
|
vulkan-utility-libraries
|
|
];
|
|
|
|
tools = with pkgs; [
|
|
vulkan-tools
|
|
wayland
|
|
wayland-protocols
|
|
libGL
|
|
];
|
|
|
|
nvidiaOffloadScript = pkgs.writeShellScriptBin "nvidia-offload" ''
|
|
export LIBVA_DRIVER_NAME=nvidia
|
|
export GBM_BACKEND=nvidia-drm
|
|
export __GLX_VENDOR_LIBRARY_NAME=nvidia
|
|
export __GL_VRR_ALLOWED=1
|
|
export XDG_SESSION_TYPE=wayland
|
|
export NVD_BACKEND=direct
|
|
export ELECTRON_OZONE_PLATFORM_HINT=auto
|
|
exec "$@"
|
|
'';
|
|
|
|
hasVendor = vendor: builtins.elem vendor cfg.vendors;
|
|
in {
|
|
options.nixos.hardware.graphics = {
|
|
enable = mkEnableOption "Enable general graphics support";
|
|
|
|
vendors = mkOption {
|
|
type = types.listOf (types.enum ["amd" "intel" "nvidia"]);
|
|
default = ["amd"];
|
|
description = "List of GPU vendors to configure support for.";
|
|
};
|
|
|
|
nvidia = {
|
|
open.enable = mkEnableOption "Enable NVidia open drivers";
|
|
|
|
package = mkOption {
|
|
type = types.enum ["stable" "beta" "production" "latest"];
|
|
default = "stable";
|
|
description = "NVidia driver package to use.";
|
|
};
|
|
};
|
|
};
|
|
|
|
config = mkIf cfg.enable (mkMerge [
|
|
{
|
|
hardware.graphics = {
|
|
enable = true;
|
|
enable32Bit = true;
|
|
extraPackages = flatten (concatMap (
|
|
vendor:
|
|
if vendor == "amd"
|
|
then commonPackages ++ mesaVulkanPackages
|
|
else if vendor == "intel"
|
|
then
|
|
commonPackages
|
|
++ mesaVulkanPackages
|
|
++ (with pkgs; [
|
|
vpl-gpu-rt
|
|
intel-media-driver
|
|
intel-compute-runtime
|
|
])
|
|
else if vendor == "nvidia"
|
|
then
|
|
commonPackages
|
|
++ (with pkgs; [
|
|
nvidiaOffloadScript
|
|
intel-media-driver
|
|
nvidia-vaapi-driver
|
|
vulkan-tools
|
|
])
|
|
else []
|
|
)
|
|
cfg.vendors);
|
|
|
|
extraPackages32 = flatten (concatMap (_: commonPackages32) cfg.vendors);
|
|
};
|
|
|
|
environment.systemPackages = flatten (concatMap (
|
|
vendor:
|
|
if vendor == "amd"
|
|
then
|
|
tools
|
|
++ (with pkgs; [
|
|
# rocmPackages.rpp
|
|
# rocmPackages.clr
|
|
])
|
|
else if vendor == "intel"
|
|
then tools
|
|
else if vendor == "nvidia"
|
|
then with pkgs; [egl-wayland libGL]
|
|
else []
|
|
)
|
|
cfg.vendors);
|
|
}
|
|
|
|
(mkIf (hasVendor "amd") {
|
|
hardware.amdgpu.overdrive.enable = true;
|
|
})
|
|
|
|
(mkIf (hasVendor "nvidia") {
|
|
hardware.nvidia = {
|
|
package =
|
|
if cfg.nvidia.package == "beta"
|
|
then config.boot.kernelPackages.nvidiaPackages.beta
|
|
else if cfg.nvidia.package == "latest"
|
|
then config.boot.kernelPackages.nvidiaPackages.latest
|
|
else if cfg.nvidia.package == "production"
|
|
then config.boot.kernelPackages.nvidiaPackages.production
|
|
else config.boot.kernelPackages.nvidiaPackages.stable;
|
|
|
|
modesetting.enable = true;
|
|
powerManagement.enable = false;
|
|
powerManagement.finegrained = false;
|
|
open = cfg.nvidia.open.enable;
|
|
nvidiaSettings = true;
|
|
};
|
|
})
|
|
]);
|
|
}
|