From 31ceb42c573a8a51d77f87f25e2847c7f8c07d82 Mon Sep 17 00:00:00 2001
From: Collin Arnett <collin@arnett.it>
Date: Fri, 12 Dec 2025 10:32:16 -0500
Subject: [PATCH] libtorch-bin: add missing runtime dependencies

---
 .../libraries/science/math/libtorch/bin.nix   | 24 +++++++++++++++----
 1 file changed, 20 insertions(+), 4 deletions(-)

diff --git a/pkgs/development/libraries/science/math/libtorch/bin.nix b/pkgs/development/libraries/science/math/libtorch/bin.nix
index 7430977674dd..f0b5a08af427 100644
--- a/pkgs/development/libraries/science/math/libtorch/bin.nix
+++ b/pkgs/development/libraries/science/math/libtorch/bin.nix
@@ -3,15 +3,17 @@
   stdenv,
   fetchzip,
   lib,
-  libcxx,
   llvmPackages,
   config,
 
-  addDriverRunpath,
+  autoAddDriverRunpath,
+  autoPatchelfHook,
   patchelf,
   fixDarwinDylibNames,
 
   cudaSupport ? config.cudaSupport,
+  cudaPackages_13,
+  libz,
 }:
 
 let
@@ -35,7 +37,11 @@ stdenv.mkDerivation {
     if stdenv.hostPlatform.isDarwin then
       [ fixDarwinDylibNames ]
     else
-      [ patchelf ] ++ lib.optionals cudaSupport [ addDriverRunpath ];
+      [
+        patchelf
+        autoPatchelfHook
+      ]
+      ++ lib.optionals cudaSupport [ autoAddDriverRunpath ];
 
   dontBuild = true;
   dontConfigure = true;
@@ -63,7 +69,17 @@ stdenv.mkDerivation {
 
   postFixup =
     let
-      rpath = lib.makeLibraryPath [ stdenv.cc.cc ];
+      rpath = lib.makeLibraryPath (
+        [ stdenv.cc.cc ]
+        ++ lib.optionals cudaSupport [
+          cudaPackages_13.cuda_cudart # libcuda.so
+          cudaPackages_13.libcufft
+          cudaPackages_13.libcurand
+          cudaPackages_13.libcusolver
+          cudaPackages_13.libcusparse
+          libz
+        ]
+      );
     in
     lib.optionalString stdenv.hostPlatform.isLinux ''
       find $out/lib -type f \( -name '*.so' -or -name '*.so.*' \) | while read lib; do