diff --git a/hardware/amd.nix b/hardware/amd.nix index 626fac4..ca5fda0 100644 --- a/hardware/amd.nix +++ b/hardware/amd.nix @@ -2,6 +2,7 @@ with lib; let cfg = config.aviallon.hardware.amd; + devCfg = config.aviallon.developer; generalCfg = config.aviallon.general; in { @@ -24,9 +25,16 @@ in "amdgpu.mes" = mkIf generalCfg.unsafeOptimizations 1; }; - environment.systemPackages = with pkgs; [ - rocm-smi - ]; + environment.systemPackages = with pkgs; [] + ++ [ + 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 = [] ++ optional cfg.useProprietary "amdgpu-pro" @@ -35,6 +43,7 @@ in "radeon" ]; + hardware.opengl.enable = true; hardware.opengl.extraPackages = with pkgs; mkIf (!cfg.useProprietary) [ rocm-opencl-icd rocm-opencl-runtime