1 files changed, 42 insertions, 0 deletions
diff --git a/modules/amd/default.nix b/modules/amd/default.nix
new file mode 100644
index 0000000..c4842e6
--- /dev/null
+++ b/modules/amd/default.nix
@@ -0,0 +1,42 @@
+{ config, pkgs, lib, ... }:
+{
+  boot = {
+    initrd = {
+      kernelModules = [ "amdgpu" ];
+    };
+  };
+
+  environment.systemPackages = with pkgs; [
+    rocmPackages.rocminfo
+    rocm.Packages.rocm-smi
+    rocmPackages.hipcc  
+  ];
+
+  hardware = {
+    graphics = {
+      enable = true;
+      extraPackages = with pkgs; [
+        rocmPackages.clr
+      ];
+    };
+  };
+  
+  nixpkgs.config = {
+    allowUnfree = lib.mkForce true;
+    rocmSupport = true;
+    rocmTargets = [ "gfx1031" ];  
+  };
+  
+  services = {
+    ollama = {
+      environmentVariables = {
+        HCC_AMDGPU_TARGET = "gfx1031";
+      };
+      rocmOverrideGfx = "10.3.0";
+    };
+
+    xserver.videoDrivers = [
+      "amdgpu"
+    ];
+  };
+}