pkgs/development/cuda-modules/packages/nvpl_tensor.nix


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

{ buildRedist, nvpl_blas }:
buildRedist {
  redistName = "nvpl";
  pname = "nvpl_tensor";

  outputs = [
    "out"
    "dev"
    "include"
    "lib"
  ];

  buildInputs = [ nvpl_blas ];

  meta = {
    description = "Part of NVIDIA Performance Libraries that provides tensor primitives";
    homepage = "https://developer.nvidia.com/nvpl";
    changelog = "https://docs.nvidia.com/nvpl/latest/tensor/release_notes.html";
  };
}