146 lines
3.7 KiB
Nix
146 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
|
|
];
|
|
|
|
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
|
|
];
|
|
|
|
nvidiaOffloadScript = pkgs.writeShellScriptBin "nvidia-offload" ''
|
|
export __NV_PRIME_RENDER_OFFLOAD=1
|
|
export __NV_PRIME_RENDER_OFFLOAD_PROVIDER=NVIDIA-G0
|
|
export __GLX_VENDOR_LIBRARY_NAME=nvidia
|
|
export __VK_LAYER_NV_optimus=NVIDIA_only
|
|
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 = mkOption {
|
|
type = lib.types.bool;
|
|
default = false;
|
|
description = "Use nvidia open driver";
|
|
};
|
|
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
|
|
intel-vaapi-driver
|
|
])
|
|
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;
|
|
nvidiaSettings = true;
|
|
};
|
|
})
|
|
]);
|
|
}
|