nix-configuration/modules/system/gpu/amd.nix

# Enables AMD GPU support.
{
  pkgs,
  config,
  lib,
  ...
}:
let
  cfg = config.aux.system.gpu.amd;
in
{
  options = {
    aux.system.gpu.amd.enable = lib.mkEnableOption "Enables AMD GPU support.";
  };

  config = lib.mkIf cfg.enable {
    boot.initrd.kernelModules = [ "amdgpu" ];
    services.xserver.videoDrivers = [ "amdgpu" ];

    hardware.opengl = {
      extraPackages = [ pkgs.amdvlk ];
      # 32-bit application compatibility
      driSupport32Bit = true;
      extraPackages32 = with pkgs; [ driversi686Linux.amdvlk ];
    };
  };
}