From 06c5b53ca56092bc65c2f6f30e31c8eeb87cbb7e Mon Sep 17 00:00:00 2001
From: haras <social@hrvojej.anonaddy.me>
Date: Mon, 16 Sep 2024 11:12:23 +0200
Subject: [PATCH] gpt4all-cuda: fix CUDA initialization

---
 pkgs/by-name/gp/gpt4all/package.nix | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/pkgs/by-name/gp/gpt4all/package.nix b/pkgs/by-name/gp/gpt4all/package.nix
index 92fccff0dfe0..7d4ecab1b7a2 100644
--- a/pkgs/by-name/gp/gpt4all/package.nix
+++ b/pkgs/by-name/gp/gpt4all/package.nix
@@ -11,6 +11,7 @@
 , wayland
 , cudaSupport ? config.cudaSupport
 , cudaPackages ? { }
+, autoAddDriverRunpath
 }:
 
 stdenv.mkDerivation (finalAttrs: {
@@ -41,6 +42,7 @@ stdenv.mkDerivation (finalAttrs: {
     qt6.wrapQtAppsHook
   ] ++ lib.optionals cudaSupport [
     cudaPackages.cuda_nvcc
+    autoAddDriverRunpath
   ];
 
   buildInputs = [
@@ -56,12 +58,13 @@ stdenv.mkDerivation (finalAttrs: {
     vulkan-headers
     wayland
   ] ++ lib.optionals cudaSupport (
-      with cudaPackages;
-      [
-        cuda_cccl
-        cuda_cudart
-        libcublas
-      ]);
+    with cudaPackages;
+    [
+      cuda_cccl
+      cuda_cudart
+      libcublas
+    ]
+  );
 
   cmakeFlags = [
     "-DKOMPUTE_OPT_USE_BUILT_IN_VULKAN_HEADER=OFF"