From 1efbc6b0643751480a879d94600b589b95e82e21 Mon Sep 17 00:00:00 2001
From: Someone Serge <sergei.kozlukov@aalto.fi>
Date: Tue, 26 Dec 2023 17:26:22 +0000
Subject: [PATCH] nix: add the impure driver's location to the DT_RUNPATHs

---
 .devops/nix/package.nix | 20 ++++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/.devops/nix/package.nix b/.devops/nix/package.nix
index cb1e8f480..c54a7c3c6 100644
--- a/.devops/nix/package.nix
+++ b/.devops/nix/package.nix
@@ -124,12 +124,20 @@ effectiveStdenv.mkDerivation (
       substituteInPlace ./*.py --replace "/usr/bin/env python" "${llama-python}/bin/python"
     '';
 
-    nativeBuildInputs = [
-      cmake
-      ninja
-      pkg-config
-      git
-    ] ++ optionals useCuda [ cudaPackages.cuda_nvcc ];
+    nativeBuildInputs =
+      [
+        cmake
+        ninja
+        pkg-config
+        git
+      ]
+      ++ optionals useCuda [
+        cudaPackages.cuda_nvcc
+
+        # TODO: Replace with autoAddDriverRunpath
+        # once https://github.com/NixOS/nixpkgs/pull/275241 has been merged
+        cudaPackages.autoAddOpenGLRunpathHook
+      ];
 
     buildInputs =
       [ mpi ]