elpa: fix CUDA build, use cudaPackages instead of cudatoolkit

This commit is contained in:
Markus Kowalewski
2023-12-26 14:03:27 +01:00
parent e785198679
commit 43d728ff88
+7 -3
View File
@@ -11,7 +11,7 @@
, enableCuda ? config.cudaSupport
# type of GPU architecture
, nvidiaArch ? "sm_60"
, cudatoolkit
, cudaPackages
} :
assert blas.isILP64 == lapack.isILP64;
@@ -43,10 +43,14 @@ stdenv.mkDerivation rec {
outputs = [ "out" "doc" "man" "dev" ];
nativeBuildInputs = [ autoreconfHook perl ];
nativeBuildInputs = [ autoreconfHook perl ]
++ lib.optionals enableCuda [ cudaPackages.cuda_nvcc ];
buildInputs = [ mpi blas lapack scalapack ]
++ lib.optional enableCuda cudatoolkit;
++ lib.optionals enableCuda [
cudaPackages.cuda_cudart
cudaPackages.libcublas
];
preConfigure = ''
export FC="mpifort"