cudaPackages.saxpy: 1/3 the runtime closure

(cherry picked from commit 3e003a17ba)
This commit is contained in:
Someone Serge 2024-01-06 23:12:29 +00:00
parent a6671578ff
commit f4474ec730
No known key found for this signature in database
GPG Key ID: 7B0E3B1390D61DA4

View File

@ -36,7 +36,9 @@ backendStdenv.mkDerivation {
buildInputs =
lib.optionals (lib.versionOlder cudaVersion "11.4") [cudatoolkit]
++ lib.optionals (lib.versionAtLeast cudaVersion "11.4") [
libcublas
libcublas.dev
libcublas.lib
libcublas.static
cuda_cudart
]
++ lib.optionals (lib.versionAtLeast cudaVersion "12.0") [cuda_cccl];