tensile: init at 4.34.0-5.3.1
This commit is contained in:
parent
9acda2f49d
commit
12b18173da
2 changed files with 36 additions and 0 deletions
34
pkgs/development/libraries/tensile/default.nix
Normal file
34
pkgs/development/libraries/tensile/default.nix
Normal file
|
@ -0,0 +1,34 @@
|
|||
{ lib
|
||||
, stdenv
|
||||
, fetchFromGitHub
|
||||
, buildPythonPackage
|
||||
, pyyaml
|
||||
, msgpack
|
||||
, pandas
|
||||
}:
|
||||
|
||||
buildPythonPackage rec {
|
||||
pname = "tensile";
|
||||
rocmVersion = "5.3.1";
|
||||
version = "4.34.0-${rocmVersion}";
|
||||
|
||||
src = fetchFromGitHub {
|
||||
owner = "ROCmSoftwarePlatform";
|
||||
repo = "Tensile";
|
||||
rev = "rocm-${rocmVersion}";
|
||||
hash = "sha256-QWt/zzBrZKM8h3MTnbLX4vN3p6cCQvo67U1C2yqAQxw=";
|
||||
};
|
||||
|
||||
buildInputs = [
|
||||
pyyaml
|
||||
msgpack
|
||||
pandas
|
||||
];
|
||||
|
||||
meta = with lib; {
|
||||
description = "GEMMs and tensor contractions";
|
||||
homepage = "https://github.com/ROCmSoftwarePlatform/Tensile";
|
||||
license = with licenses; [ mit ];
|
||||
maintainers = with maintainers; [ Madouura ];
|
||||
};
|
||||
}
|
|
@ -14889,6 +14889,8 @@ with pkgs;
|
|||
|
||||
rocfft = callPackage ../development/libraries/rocfft { };
|
||||
|
||||
tensile = python3Packages.callPackage ../development/libraries/tensile { };
|
||||
|
||||
rtags = callPackage ../development/tools/rtags {
|
||||
inherit (darwin) apple_sdk;
|
||||
};
|
||||
|
|
Loading…
Reference in a new issue