pyrox.dev
cudaPackages.cutensor: 1.3.1.3 -> 1.5.0.3
Update cutensor to the latest version compatible with its only dependent
in nixpkgs, cupy. Future work might wish to make multiple versions
available. Also update the URL format for all known versions.
+4
-1
pkgs/development/libraries/science/math/cutensor/generic.nix
+4
-1
pkgs/development/libraries/science/math/cutensor/generic.nix
···
13
13
let
14
14
mostOfVersion = builtins.concatStringsSep "."
15
15
(lib.take 3 (lib.versions.splitVersion version));
16
+
platform = "${stdenv.hostPlatform.parsed.kernel.name}-${stdenv.hostPlatform.parsed.cpu.name}";
16
17
in
17
18
18
19
stdenv.mkDerivation {
···
20
21
inherit version;
21
22
22
23
src = fetchurl {
23
-
url = "https://developer.download.nvidia.com/compute/cutensor/${mostOfVersion}/local_installers/libcutensor-${stdenv.hostPlatform.parsed.kernel.name}-${stdenv.hostPlatform.parsed.cpu.name}-${version}.tar.gz";
24
+
url = if lib.versionOlder mostOfVersion "1.3.3"
25
+
then "https://developer.download.nvidia.com/compute/cutensor/${mostOfVersion}/local_installers/libcutensor-${platform}-${version}.tar.gz"
26
+
else "https://developer.download.nvidia.com/compute/cutensor/redist/libcutensor/${platform}/libcutensor-${platform}-${version}-archive.tar.xz";
24
27
inherit hash;
25
28
};
26
29
+3
-3
pkgs/top-level/cuda-packages.nix
+3
-3
pkgs/top-level/cuda-packages.nix
···
28
28
"1.2.2.5" = {
29
29
hash = "sha256-lU7iK4DWuC/U3s1Ct/rq2Gr3w4F2U7RYYgpmF05bibY=";
30
30
};
31
-
"1.3.1.3" = {
32
-
hash = "sha256-mNlVnabB2IC3HnYY0mb06RLqQzDxN9ePGVeBy3hkBC8=";
31
+
"1.5.0.3" = {
32
+
hash = "sha256-T96+lPC6OTOkIs/z3QWg73oYVSyidN0SVkBWmT9VRx0=";
33
33
};
34
34
};
35
35
36
36
inherit (final) cudaMajorMinorVersion cudaMajorVersion;
37
37
38
38
cutensor = buildCuTensorPackage rec {
39
-
version = if cudaMajorMinorVersion == "10.1" then "1.2.2.5" else "1.3.1.3";
39
+
version = if cudaMajorMinorVersion == "10.1" then "1.2.2.5" else "1.5.0.3";
40
40
inherit (cuTensorVersions.${version}) hash;
41
41
# This can go into generic.nix
42
42
libPath = "lib/${if cudaMajorVersion == "10" then cudaMajorMinorVersion else cudaMajorVersion}";