diff options
author | Markus Kowalewski <markus.kowalewski@gmail.com> | 2024-01-08 20:26:09 +0100 |
---|---|---|
committer | Markus Kowalewski <markus.kowalewski@gmail.com> | 2024-01-08 20:26:09 +0100 |
commit | 8156485148b0c9080e5fc22153a038aa4a096245 (patch) | |
tree | 3ff5bb7b96057fa47815c60ca96d84590a7d913d /pkgs/by-name/si | |
parent | 3bd19bc4ef76826274bd0ef7d0899fa97d8990cb (diff) |
sirius: fix CUDA build
Diffstat (limited to 'pkgs/by-name/si')
-rw-r--r-- | pkgs/by-name/si/sirius/package.nix | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/pkgs/by-name/si/sirius/package.nix b/pkgs/by-name/si/sirius/package.nix index 8518092203b8e..ee6052cf09c16 100644 --- a/pkgs/by-name/si/sirius/package.nix +++ b/pkgs/by-name/si/sirius/package.nix @@ -55,7 +55,7 @@ stdenv.mkDerivation rec { cmake gfortran pkg-config - ]; + ] ++ lib.optional (gpuBackend == "cuda") cudaPackages.cuda_nvcc; buildInputs = [ blas @@ -72,8 +72,12 @@ stdenv.mkDerivation rec { eigen libvdwxc ] - ++ lib.optional (gpuBackend == "cuda") cudaPackages.cudatoolkit - ++ lib.optionals (gpuBackend == "rocm") [ + ++ lib.optionals (gpuBackend == "cuda") [ + cudaPackages.cuda_cudart + cudaPackages.cuda_profiler_api + cudaPackages.cudatoolkit + cudaPackages.libcublas + ] ++ lib.optionals (gpuBackend == "rocm") [ rocmPackages.clr rocmPackages.rocblas ] ++ lib.optional stdenv.isDarwin llvmPackages.openmp |