From 33905abff4d1a51ddd53866c23a30536b35b7c50 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= <me@danieldk.eu>
Date: Mon, 29 Sep 2025 15:08:04 +0000
Subject: [PATCH] ROCm: add rocwmma-devel

---
 lib/torch-extension/default.nix | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/lib/torch-extension/default.nix b/lib/torch-extension/default.nix
index 41e90860..4cc583dc 100644
--- a/lib/torch-extension/default.nix
+++ b/lib/torch-extension/default.nix
@@ -130,7 +130,13 @@ stdenv.mkDerivation (prevAttrs: {
       libcusparse
     ]
   )
-  ++ lib.optionals rocmSupport (with rocmPackages; [ hipsparselt ])
+  ++ lib.optionals rocmSupport (
+    with rocmPackages;
+    [
+      hipsparselt
+      rocwmma-devel
+    ]
+  )
   ++ lib.optionals xpuSupport ([
     oneapi-torch-dev
     onednn-xpu