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 = "cutensor";
10 pname = "libcutensor";
11
12 outputs = [
13 "out"
14 "dev"
15 "include"
16 "lib"
17 "static"
18 ];
19
20 allowFHSReferences = true;
21
22 buildInputs = [
23 (lib.getLib libcublas)
24 ]
25 # For some reason, the 1.4.x release of cuTENSOR requires the cudart library.
26 ++ lib.optionals (lib.hasPrefix "1.4" finalAttrs.version) [ (lib.getLib cuda_cudart) ];
27
28 meta = {
29 description = "GPU-accelerated tensor linear algebra library for tensor contraction, reduction, and elementwise operations";
30 longDescription = ''
31 NVIDIA cuTENSOR is a GPU-accelerated tensor linear algebra library for tensor contraction, reduction, and
32 elementwise operations. Using cuTENSOR, applications can harness the specialized tensor cores on NVIDIA GPUs for
33 high-performance tensor computations and accelerate deep learning training and inference, computer vision,
34 quantum chemistry, and computational physics workloads.
35 '';
36 homepage = "https://developer.nvidia.com/cutensor";
37 changelog = "https://docs.nvidia.com/cuda/cutensor/latest/release_notes.html";
38
39 license = _cuda.lib.licenses.cutensor;
40 };
41})