dev/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

gpt4all-cuda: fix CUDA initialization (#342251)

Browse Source
This commit is contained in:
Fabián Heredia Montiel 2024-09-17 09:18:07 -06:00 committed by GitHub
commit e2f071fe68
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pkgs/by-name/gp/gpt4all/package.nix
View File

@ -11,6 +11,7 @@
, wayland , wayland
, cudaSupport ? config.cudaSupport , cudaSupport ? config.cudaSupport
, cudaPackages ? { } , cudaPackages ? { }
, autoAddDriverRunpath
}: }:
stdenv.mkDerivation (finalAttrs: { stdenv.mkDerivation (finalAttrs: {
@ -41,6 +42,7 @@ stdenv.mkDerivation (finalAttrs: {
qt6.wrapQtAppsHook qt6.wrapQtAppsHook
] ++ lib.optionals cudaSupport [ ] ++ lib.optionals cudaSupport [
cudaPackages.cuda_nvcc cudaPackages.cuda_nvcc
autoAddDriverRunpath
]; ];
buildInputs = [ buildInputs = [
@ -56,12 +58,13 @@ stdenv.mkDerivation (finalAttrs: {
vulkan-headers vulkan-headers
wayland wayland
] ++ lib.optionals cudaSupport ( ] ++ lib.optionals cudaSupport (
with cudaPackages; with cudaPackages;
[ [
cuda_cccl cuda_cccl
cuda_cudart cuda_cudart
libcublas libcublas
]); ]
);
cmakeFlags = [ cmakeFlags = [
"-DKOMPUTE_OPT_USE_BUILT_IN_VULKAN_HEADER=OFF" "-DKOMPUTE_OPT_USE_BUILT_IN_VULKAN_HEADER=OFF"