31 lines
682 B
Nix
31 lines
682 B
Nix
{ config, lib, pkgs, ... }:
|
|
|
|
with lib;
|
|
let
|
|
cfg = config.modules.system.hardware.graphics.amdgpu;
|
|
in
|
|
{
|
|
options.modules.system.hardware.graphics.amdgpu = {
|
|
enable = mkEnableOption ''
|
|
Enable amdgpu with my custom configurations
|
|
'';
|
|
};
|
|
config = mkIf cfg.enable {
|
|
boot.initrd.kernelModules = [ "amdgpu" ];
|
|
|
|
# Set acceleration to rocm
|
|
services.ollama.acceleration = "rocm";
|
|
|
|
# Load amdgpu driver for Xorg and Wayland
|
|
services.xserver = {
|
|
enable = true;
|
|
videoDrivers = ["amdgpu"];
|
|
};
|
|
|
|
hardware.graphics = {
|
|
enable = true;
|
|
enable32Bit = true;
|
|
extraPackages = with pkgs; [ rocmPackages.clr.icd ];
|
|
};
|
|
};
|
|
}
|