{ 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"
    ];
  };
}