tangled
alpha
login
or
join now
pyrox.dev
/
nixpkgs
0
fork
atom
lol
0
fork
atom
overview
issues
pulls
pipelines
tensile: init at 4.34.0-5.3.1
Madoura
3 years ago
12b18173
9acda2f4
+36
2 changed files
expand all
collapse all
unified
split
pkgs
development
libraries
tensile
default.nix
top-level
all-packages.nix
+34
pkgs/development/libraries/tensile/default.nix
···
1
1
+
{ lib
2
2
+
, stdenv
3
3
+
, fetchFromGitHub
4
4
+
, buildPythonPackage
5
5
+
, pyyaml
6
6
+
, msgpack
7
7
+
, pandas
8
8
+
}:
9
9
+
10
10
+
buildPythonPackage rec {
11
11
+
pname = "tensile";
12
12
+
rocmVersion = "5.3.1";
13
13
+
version = "4.34.0-${rocmVersion}";
14
14
+
15
15
+
src = fetchFromGitHub {
16
16
+
owner = "ROCmSoftwarePlatform";
17
17
+
repo = "Tensile";
18
18
+
rev = "rocm-${rocmVersion}";
19
19
+
hash = "sha256-QWt/zzBrZKM8h3MTnbLX4vN3p6cCQvo67U1C2yqAQxw=";
20
20
+
};
21
21
+
22
22
+
buildInputs = [
23
23
+
pyyaml
24
24
+
msgpack
25
25
+
pandas
26
26
+
];
27
27
+
28
28
+
meta = with lib; {
29
29
+
description = "GEMMs and tensor contractions";
30
30
+
homepage = "https://github.com/ROCmSoftwarePlatform/Tensile";
31
31
+
license = with licenses; [ mit ];
32
32
+
maintainers = with maintainers; [ Madouura ];
33
33
+
};
34
34
+
}
+2
pkgs/top-level/all-packages.nix
···
14889
14889
14890
14890
rocfft = callPackage ../development/libraries/rocfft { };
14891
14891
14892
14892
+
tensile = python3Packages.callPackage ../development/libraries/tensile { };
14893
14893
+
14892
14894
rtags = callPackage ../development/tools/rtags {
14893
14895
inherit (darwin) apple_sdk;
14894
14896
};