tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
1{
2 _cuda,
3 buildRedist,
4 cuda_cudart,
5 lib,
6 libcublas,
7}:
8buildRedist (finalAttrs: {
9 redistName = "cusparselt";
10 pname = "libcusparse_lt";
11
12 outputs = [
13 "out"
14 "dev"
15 "include"
16 "lib"
17 "static"
18 ];
19
20 buildInputs = [
21 (lib.getLib libcublas)
22 ]
23 # For some reason, the 1.4.x release of cusparselt requires the cudart library.
24 ++ lib.optionals (lib.hasPrefix "1.4" finalAttrs.version) [ (lib.getLib cuda_cudart) ];
25
26 meta = {
27 description = "High-performance CUDA library dedicated to general matrix-matrix operations in which at least one operand is a structured sparse matrix with 50% sparsity ratio";
28 longDescription = ''
29 NVIDIA cuSPARSELt is a high-performance CUDA library dedicated to general matrix-matrix operations in which at
30 least one operand is a structured sparse matrix with 50% sparsity ratio.
31 '';
32 homepage = "https://developer.nvidia.com/cusparselt-downloads";
33 changelog = "https://docs.nvidia.com/cuda/cublasmp/release_notes";
34
35 maintainers = [ lib.maintainers.sepiabrown ];
36 license = _cuda.lib.licenses.cusparse_lt;
37 };
38})