pyrox.dev
1{
2 cuda_cudart,
3 lib,
4 libcublas,
5}:
6finalAttrs: prevAttrs: {
7 buildInputs =
8 prevAttrs.buildInputs or [ ]
9 ++ [ (lib.getLib libcublas) ]
10 # For some reason, the 1.4.x release of cusparselt requires the cudart library.
11 ++ lib.optionals (lib.hasPrefix "1.4" finalAttrs.version) [ (lib.getLib cuda_cudart) ];
12 meta = prevAttrs.meta or { } // {
13 description = "cuSPARSELt: A High-Performance CUDA Library for Sparse Matrix-Matrix Multiplication";
14 homepage = "https://developer.nvidia.com/cusparselt-downloads";
15 maintainers = prevAttrs.meta.maintainers or [ ] ++ [ lib.maintainers.sepiabrown ];
16 teams = prevAttrs.meta.teams or [ ];
17 license = lib.licenses.unfreeRedistributable // {
18 shortName = "cuSPARSELt EULA";
19 fullName = "cuSPARSELt SUPPLEMENT TO SOFTWARE LICENSE AGREEMENT FOR NVIDIA SOFTWARE DEVELOPMENT KITS";
20 url = "https://docs.nvidia.com/cuda/cusparselt/license.html";
21 };
22 };
23}