tensile: init at 4.34.0-5.3.1

This commit is contained in:
Madoura 2022-10-25 06:44:37 -05:00
parent 9acda2f49d
commit 12b18173da
No known key found for this signature in database
GPG key ID: 3201136B3DB072F9
2 changed files with 36 additions and 0 deletions

View 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 ];
};
}

View file

@ -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;
};