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 cuTENSOR requires the cudart library.
11 ++ lib.optionals (lib.hasPrefix "1.4" finalAttrs.version) [ (lib.getLib cuda_cudart) ];
12 meta = prevAttrs.meta or { } // {
13 description = "cuTENSOR: A High-Performance CUDA Library For Tensor Primitives";
14 homepage = "https://developer.nvidia.com/cutensor";
15 maintainers = prevAttrs.meta.maintainers or [ ] ++ [ lib.maintainers.obsidian-systems-maintenance ];
16 teams = prevAttrs.meta.teams;
17 license = lib.licenses.unfreeRedistributable // {
18 shortName = "cuTENSOR EULA";
19 fullName = "cuTENSOR SUPPLEMENT TO SOFTWARE LICENSE AGREEMENT FOR NVIDIA SOFTWARE DEVELOPMENT KITS";
20 url = "https://docs.nvidia.com/cuda/cutensor/license.html";
21 };
22 };
23}