aviallon/nixos-lib
1
0
Fork 0
mirror of https://github.com/aviallon/nixos-lib.git synced 2026-04-06 01:38:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Hardware/AMD] Begin rocm-overlay suport

Browse source
This commit is contained in:
Antoine Viallon 2022-10-12 18:50:45 +02:00
parent 3dc72fae02
commit 1e54cd710b
Signed by: aviallon
GPG key ID: 186FC35EDEB25716
1 changed files with 12 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

13
hardware/amd.nix
View file

@ -2,6 +2,7 @@
with lib; with lib;
let let
cfg = config.aviallon.hardware.amd; cfg = config.aviallon.hardware.amd;
devCfg = config.aviallon.developer;
generalCfg = config.aviallon.general; generalCfg = config.aviallon.general;
in in
{ {
@ -24,9 +25,16 @@ in
"amdgpu.mes" = mkIf generalCfg.unsafeOptimizations 1; "amdgpu.mes" = mkIf generalCfg.unsafeOptimizations 1;
}; };
environment.systemPackages = with pkgs; [ environment.systemPackages = with pkgs; []
++ [
rocm-smi rocm-smi
]; ]
++ optionals devCfg.enable ([]
++ [ rocminfo ]
++ optional (hasAttr "hipify-perl" pkgs) pkgs.hipify-perl
++ optional (hasAttr "tensorflow2-rocm" pkgs) pkgs.tensorflow2-rocm
)
;
services.xserver.videoDrivers = [] services.xserver.videoDrivers = []
++ optional cfg.useProprietary "amdgpu-pro" ++ optional cfg.useProprietary "amdgpu-pro"
@ -35,6 +43,7 @@ in
"radeon" "radeon"
]; ];
hardware.opengl.enable = true;
hardware.opengl.extraPackages = with pkgs; mkIf (!cfg.useProprietary) [ hardware.opengl.extraPackages = with pkgs; mkIf (!cfg.useProprietary) [
rocm-opencl-icd rocm-opencl-icd
rocm-opencl-runtime rocm-opencl-runtime