From 4107c197699bc92e0c2127dc2771a37d47ed9a7c Mon Sep 17 00:00:00 2001
From: Dmitry Voronin <account@voronind.com>
Date: Wed, 20 Mar 2024 22:33:37 +0300
Subject: [PATCH] AmdGpu : Enable ROCm support.

---
 module/AmdGpu.nix | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/module/AmdGpu.nix b/module/AmdGpu.nix
index 122f410..992a1ac 100644
--- a/module/AmdGpu.nix
+++ b/module/AmdGpu.nix
@@ -1,9 +1,19 @@
-{ pkgs, ... }: {
+{ nixpkgs, pkgs, ... }: {
 	boot.initrd.kernelModules     = [ "amdgpu" ];
 	services.xserver.videoDrivers = [ "amdgpu" ];
 	hardware.opengl.driSupport      = true;
 	hardware.opengl.driSupport32Bit = true;
 	environment.variables.AMD_VULKAN_ICD = "RADV";
+
+	# Compute.
+	nixpkgs.config.rocmSupport = true;
+	systemd.tmpfiles.rules = [
+		"L+    /opt/rocm/hip   -    -    -     -    ${pkgs.rocmPackages.clr}"
+	];
+	hardware.opengl.extraPackages = with pkgs; [
+		rocmPackages.clr.icd
+	];
+
 	# AMDVLK was broken for me (huge stuttering). So keep it disabled, at least for now.
 	# hardware.opengl.extraPackages = with pkgs; [
 	# 	amdvlk