646b892680
Some checks failed
Periodic Merges (6h) / master → staging-nixos (push) Failing after 12m50s
Periodic Merges (6h) / master → staging-next (push) Failing after 12m54s
Periodic Merges (24h) / merge-base(master,staging) → haskell-updates (push) Failing after 11m54s
Periodic Merges (6h) / staging-next → staging (push) Failing after 12m13s
Periodic Merges (24h) / staging-next-25.05 → staging-25.05 (push) Failing after 13m24s
Periodic Merges (24h) / release-25.05 → staging-next-25.05 (push) Failing after 14m28s
24 lines
914 B
Nix
24 lines
914 B
Nix
{
|
|
cuda_cudart,
|
|
lib,
|
|
libcublas,
|
|
}:
|
|
finalAttrs: prevAttrs: {
|
|
buildInputs =
|
|
prevAttrs.buildInputs or [ ]
|
|
++ [ (lib.getLib libcublas) ]
|
|
# For some reason, the 1.4.x release of cuTENSOR requires the cudart library.
|
|
++ lib.optionals (lib.hasPrefix "1.4" finalAttrs.version) [ (lib.getLib cuda_cudart) ];
|
|
meta = prevAttrs.meta or { } // {
|
|
description = "cuTENSOR: A High-Performance CUDA Library For Tensor Primitives";
|
|
homepage = "https://developer.nvidia.com/cutensor";
|
|
maintainers = prevAttrs.meta.maintainers or [ ] ++ [ lib.maintainers.obsidian-systems-maintenance ];
|
|
teams = prevAttrs.meta.teams;
|
|
license = lib.licenses.unfreeRedistributable // {
|
|
shortName = "cuTENSOR EULA";
|
|
fullName = "cuTENSOR SUPPLEMENT TO SOFTWARE LICENSE AGREEMENT FOR NVIDIA SOFTWARE DEVELOPMENT KITS";
|
|
url = "https://docs.nvidia.com/cuda/cutensor/license.html";
|
|
};
|
|
};
|
|
}
|