nix: add the impure driver's location to the DT_RUNPATHs

ggerganov · Dec 26, 2023 · 1efbc6b · 1efbc6b
1 parent ae6bebc
commit 1efbc6b
Showing 1 changed file with 14 additions and 6 deletions.
diff --git 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 ]