From 22d3b0a94ef80997dfb214e6813146365a4edb3e Mon Sep 17 00:00:00 2001
From: rm-dr <96270320+rm-dr@users.noreply.github.com>
Date: Tue, 16 Dec 2025 22:11:41 -0800
Subject: [PATCH] nix

---
 shell.nix | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/shell.nix b/shell.nix
index f0a0d7d..d429588 100644
--- a/shell.nix
+++ b/shell.nix
@@ -60,6 +60,12 @@ pkgs.mkShell {
     export OPENSSL_LIB_DIR="${pkgs.openssl.out}/lib"
     export OPENSSL_INCLUDE_DIR="${pkgs.openssl.dev}/include"
 
+
+    export CUDA_PATH=${pkgs.cudatoolkit}
+    export LD_LIBRARY_PATH=${pkgs.linuxPackages.nvidia_x11}/lib:${pkgs.ncurses5}/lib
+    export EXTRA_LDFLAGS="-L/lib -L${pkgs.linuxPackages.nvidia_x11}/lib"
+    export EXTRA_CCFLAGS="-I/usr/include"
+
     # Not sure which of these are necessary...
     # For DuckDB
     export CMAKE_PREFIX_PATH="${pkgs.cmake}/bin:$CMAKE_PREFIX_PATH"