cudaPackages.saxpy: 1/3 the runtime closure
(cherry picked from commit 3e003a17bafb8ad2c9026843c241412e2d61a541)
This commit is contained in:
@@ -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];
|
||||
|
||||
Reference in New Issue
Block a user