Modularize hardware config and remove specific nix folder

2024-07-20 01:56:54 +02:00 · 2024-07-20 01:56:54 +02:00 · 9ac5b25036
commit 9ac5b25036
parent a5042bb645
72 changed files with 322 additions and 100 deletions
--- a/modules/programs/gpu.nix
+++ b/modules/programs/gpu.nix
@ -0,0 +1,63 @@
+{ lib, config, options, pkgs, ... }: {
+
+  options.mods = {
+    amdgpu.enable = lib.mkOption {
+      default = false;
+      type = lib.types.bool;
+      example = true;
+      description = ''
+        Enables amdgpu support.
+      '';
+    };
+    vapi = {
+      enable = lib.mkOption {
+        default = false;
+        type = lib.types.bool;
+        example = true;
+        description = ''
+          Enables vapi.
+        '';
+      };
+      rocm.enable = lib.mkOption {
+        default = false;
+        type = lib.types.bool;
+        example = true;
+        description = ''
+          Enables rocm support.
+        '';
+      };
+    };
+  };
+
+  config = lib.mkIf config.mods.vapi.enable
+    (lib.optionalAttrs
+      (options?hardware.graphics)
+      {
+        boot = lib.mkIf config.mods.amdgpu.enable {
+          kernelModules = [ "kvm-amd" ];
+          initrd.kernelModules = [ "amdgpu" ];
+          kernelParams = [
+            "amdgpu.ppfeaturemask=0xffffffff"
+          ];
+        };
+        hardware = {
+          graphics =
+            let
+              base_packages = [
+                pkgs.libvdpau-va-gl
+                pkgs.vaapiVdpau
+              ];
+              rocm_packages = [
+                pkgs.rocmPackages.clr.icd
+                pkgs.rocm-opencl-runtime
+              ];
+            in
+            {
+              enable = true;
+              enable32Bit = lib.mkDefault true;
+              extraPackages = base_packages ++
+                (lib.lists.optionals config.mods.vapi.rocm.enable rocm_packages);
+            };
+        };
+      });
+}