From 9b804b8941f64fc979470081e7c2048934a88c77 Mon Sep 17 00:00:00 2001
From: Break Yang <breakds@gmail.com>
Date: Sun, 4 Jun 2023 10:53:18 -0700
Subject: [PATCH] cudaPackages.cudatoolkit: mark libnvrtc-builtins needed for
 libnvrtc

libnvrtc dlopens libnvrtc-builtins without listing it in DT_NEEDED, thus it was being stripped out from the runpath
---
 pkgs/development/compilers/cudatoolkit/common.nix | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pkgs/development/compilers/cudatoolkit/common.nix b/pkgs/development/compilers/cudatoolkit/common.nix
index 2f8a8e51f4ee..55d7e2fb64cd 100644
--- a/pkgs/development/compilers/cudatoolkit/common.nix
+++ b/pkgs/development/compilers/cudatoolkit/common.nix
@@ -158,6 +158,10 @@ backendStdenv.mkDerivation rec {
     "libcom_err.so.2"
   ];
 
+  preFixup = ''
+    patchelf $out/lib64/libnvrtc.so --add-needed libnvrtc-builtins.so
+  '';
+
   unpackPhase = ''
     sh $src --keep --noexec
 
@@ -342,4 +346,3 @@ backendStdenv.mkDerivation rec {
     maintainers = teams.cuda.members;
   };
 }
-