cudaPackages.saxpy: 1/3 the runtime closure
(cherry picked from commit 3e003a17bafb8ad2c9026843c241412e2d61a541)
This commit is contained in:
parent
a6671578ff
commit
f4474ec730
1 changed files with 3 additions and 1 deletions
|
@ -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];
|
||||
|
|
Loading…
Reference in a new issue