74 lines
1.6 KiB
Nix
74 lines
1.6 KiB
Nix
{
|
|
lib,
|
|
config,
|
|
pkgs,
|
|
inputs,
|
|
pkgsVersion,
|
|
...
|
|
}:
|
|
let
|
|
cfg = config.custom.hardware.amd-gpu;
|
|
in
|
|
{
|
|
options.custom.hardware.amd-gpu = {
|
|
enable = lib.mkEnableOption "Enables amd gpus";
|
|
overdrive = lib.mkEnableOption "Enables overdrive";
|
|
rocm.enable = lib.mkEnableOption "Enables rocm";
|
|
};
|
|
|
|
config = lib.mkIf cfg.enable {
|
|
boot.kernelParams = lib.mkIf cfg.overdrive [
|
|
"amdgpu.ppfeaturemask=0xfff7ffff"
|
|
"split_lock_detect=off"
|
|
];
|
|
|
|
hardware.graphics = {
|
|
enable = true;
|
|
enable32Bit = true;
|
|
extraPackages = lib.mkIf cfg.rocm.enable (
|
|
with pkgs;
|
|
[
|
|
rocmPackages.clr
|
|
rocmPackages.clr.icd
|
|
]
|
|
);
|
|
};
|
|
|
|
hardware.amdgpu.initrd.enable = lib.mkDefault true;
|
|
services.xserver.videoDrivers = [ "amdgpu" ];
|
|
# controller (overclock, undervolt, fan curves)
|
|
environment.systemPackages =
|
|
with pkgs;
|
|
[
|
|
nvtopPackages.amd
|
|
amdgpu_top
|
|
]
|
|
++ lib.optionals cfg.rocm.enable [
|
|
clinfo
|
|
rocmPackages.rocminfo
|
|
]
|
|
++ lib.optionals cfg.overdrive [
|
|
lact
|
|
];
|
|
systemd = {
|
|
packages = lib.mkIf cfg.overdrive (with pkgs; [ lact ]);
|
|
services.lactd.wantedBy = lib.mkIf cfg.overdrive [ "multi-user.target" ];
|
|
#rocm
|
|
tmpfiles.rules =
|
|
let
|
|
rocmEnv = pkgs.symlinkJoin {
|
|
name = "rocm-combined";
|
|
paths = with pkgs.rocmPackages; [
|
|
rocblas
|
|
hipblas
|
|
clr
|
|
];
|
|
};
|
|
in
|
|
lib.mkIf cfg.rocm.enable [
|
|
"L+ /opt/rocm - - - - ${rocmEnv}"
|
|
];
|
|
};
|
|
};
|
|
}
|