From d1e4dfa05782028cd6f0eb366cc9f84da3df86ad Mon Sep 17 00:00:00 2001
From: "Alexandre Cavalheiro S. Tiago da Silva" <contact@thewizard.link>
Date: Tue, 28 Jan 2025 04:45:42 -0300
Subject: [PATCH] nvidia_oc: add driver runpath hook

This adds the `autoAddDriverRunpath` hook to `nvidia_oc`, patching the
ELF and giving the program the path where `libnvidia-ml.so` is located.
---
 pkgs/by-name/nv/nvidia_oc/package.nix | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/pkgs/by-name/nv/nvidia_oc/package.nix b/pkgs/by-name/nv/nvidia_oc/package.nix
index e5282d152240..1d6ca57286ae 100644
--- a/pkgs/by-name/nv/nvidia_oc/package.nix
+++ b/pkgs/by-name/nv/nvidia_oc/package.nix
@@ -1,7 +1,8 @@
 {
   lib,
-  fetchFromGitHub,
   rustPlatform,
+  fetchFromGitHub,
+  autoAddDriverRunpath,
   versionCheckHook,
 }:
 
@@ -19,6 +20,10 @@ rustPlatform.buildRustPackage rec {
   useFetchCargoVendor = true;
   cargoHash = "sha256-DuuHqBhL25ghgcYxcOtWRArUqL9+c3d5mBrAcWTAFW8=";
 
+  nativeBuildInputs = [
+    autoAddDriverRunpath
+  ];
+
   doInstallCheck = true;
 
   nativeInstallCheckInputs = [