From b890e55c2eecb88a6497cd392f911b20a6b37852 Mon Sep 17 00:00:00 2001
From: "J. T. Elscott" <78996170+99jte@users.noreply.github.com>
Date: Wed, 20 Aug 2025 07:31:15 -0700
Subject: [PATCH] hashcat: add rocm support

Just copy what's already done for CUDA.
---
 pkgs/by-name/ha/hashcat/package.nix | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/pkgs/by-name/ha/hashcat/package.nix b/pkgs/by-name/ha/hashcat/package.nix
index c4f8f9f6b295..20764257ffc4 100644
--- a/pkgs/by-name/ha/hashcat/package.nix
+++ b/pkgs/by-name/ha/hashcat/package.nix
@@ -12,6 +12,8 @@
   ocl-icd,
   perl,
   python3,
+  rocmPackages ? { },
+  rocmSupport ? config.rocmSupport,
   xxHash,
   zlib,
   libiconv,
@@ -43,7 +45,7 @@ stdenv.mkDerivation rec {
   nativeBuildInputs = [
     makeWrapper
   ]
-  ++ lib.optionals cudaSupport [
+  ++ lib.optionals (cudaSupport || rocmSupport) [
     addDriverRunpath
   ];
 
@@ -102,13 +104,16 @@ stdenv.mkDerivation rec {
         ++ lib.optionals cudaSupport [
           "${cudaPackages_12_4.cudatoolkit}/lib"
         ]
+        ++ lib.optionals rocmSupport [
+          "${rocmPackages.clr}/lib"
+        ]
       );
     in
     ''
       wrapProgram $out/bin/hashcat \
         --prefix LD_LIBRARY_PATH : ${lib.escapeShellArg LD_LIBRARY_PATH}
     ''
-    + lib.optionalString cudaSupport ''
+    + lib.optionalString (cudaSupport || rocmSupport) ''
       for program in $out/bin/hashcat $out/bin/.hashcat-wrapped; do
         isELF "$program" || continue
         addDriverRunpath "$program"