Unverified Commit e2f071fe authored by Fabián Heredia Montiel's avatar Fabián Heredia Montiel Committed by GitHub
Browse files

gpt4all-cuda: fix CUDA initialization (#342251)

parents 45b95421 06c5b53c
Loading
Loading
Loading
Loading
+9 −6
Original line number Diff line number Diff line
@@ -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 = [
@@ -61,7 +63,8 @@ stdenv.mkDerivation (finalAttrs: {
      cuda_cccl
      cuda_cudart
      libcublas
      ]);
    ]
  );

  cmakeFlags = [
    "-DKOMPUTE_OPT_USE_BUILT_IN_VULKAN_HEADER=OFF"