tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
1{
2 buildRedist,
3 cudaAtLeast,
4 lib,
5 libcublas,
6 libcusparse,
7 libnvjitlink,
8}:
9buildRedist {
10 redistName = "cuda";
11 pname = "libcusolver";
12
13 outputs = [
14 "out"
15 "dev"
16 "include"
17 "lib"
18 "static"
19 "stubs"
20 ];
21
22 buildInputs =
23 # Always depends on this
24 [ (lib.getLib libcublas) ]
25 # Dependency from 12.0 and on
26 ++ lib.optionals (cudaAtLeast "12.0") [ libnvjitlink ]
27 # Dependency from 12.1 and on
28 ++ lib.optionals (cudaAtLeast "12.1") [ (lib.getLib libcusparse) ];
29
30 meta = {
31 description = "Collection of dense and sparse direct linear solvers and Eigen solvers";
32 longDescription = ''
33 The NVIDIA cuSOLVER library provides a collection of dense and sparse direct linear solvers and Eigen solvers
34 which deliver significant acceleration for Computer Vision, CFD, Computational Chemistry, and Linear
35 Optimization applications.
36 '';
37 homepage = "https://developer.nvidia.com/cusolver";
38 };
39}