pkgs/development/libraries/science/math/cutensor/default.nix


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

{ callPackage
, cudatoolkit_10_1, cudatoolkit_10_2
, cudatoolkit_11_0, cudatoolkit_11_1, cudatoolkit_11_2
}:

rec {
  cutensor_cudatoolkit_10_1 = callPackage ./generic.nix rec {
    version = "1.2.2.5";
    libPath = "lib/10.1";
    cudatoolkit = cudatoolkit_10_1;
    # 1.2.2 is compatible with CUDA 11.0, 11.1, and 11.2:
    # ephemeral doc at https://developer.nvidia.com/cutensor/downloads
    sha256 = "1dl9bd71frhac9cb8lvnh71zfsnqxbxbfhndvva2zf6nh0my4klm";
  };

  cutensor_cudatoolkit_10_2 = cutensor_cudatoolkit_10_1.override {
    libPath = "lib/10.2";
    cudatoolkit = cudatoolkit_10_2;
  };

  cutensor_cudatoolkit_10 = cutensor_cudatoolkit_10_2;

  cutensor_cudatoolkit_11_0 = cutensor_cudatoolkit_10_2.override {
    libPath = "lib/11";
    cudatoolkit = cudatoolkit_11_0;
  };

  cutensor_cudatoolkit_11_1 = cutensor_cudatoolkit_11_0.override {
    cudatoolkit = cudatoolkit_11_1;
  };

  cutensor_cudatoolkit_11_2 = cutensor_cudatoolkit_11_0.override {
    cudatoolkit = cudatoolkit_11_2;
  };

  cutensor_cudatoolkit_11 = cutensor_cudatoolkit_11_2;
}