cudaPackages.saxpy: 1/3 the runtime closure

(cherry picked from commit 3e003a17bafb8ad2c9026843c241412e2d61a541)
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];