Skip to content

Commit

Permalink
tensile: init at 4.34.0-5.3.1
Browse files Browse the repository at this point in the history
  • Loading branch information
Madouura committed Nov 3, 2022
1 parent ee419e4 commit bb2286f
Show file tree
Hide file tree
Showing 2 changed files with 36 additions and 0 deletions.
34 changes: 34 additions & 0 deletions pkgs/development/libraries/tensile/default.nix
Original file line number Diff line number Diff line change
@@ -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 ];
};
}
2 changes: 2 additions & 0 deletions pkgs/top-level/all-packages.nix
Original file line number Diff line number Diff line change
Expand Up @@ -14859,6 +14859,8 @@ with pkgs;

rocminfo = callPackage ../development/tools/rocminfo { };

tensile = python3Packages.callPackage ../development/libraries/tensile { };

rtags = callPackage ../development/tools/rtags {
inherit (darwin) apple_sdk;
};
Expand Down

0 comments on commit bb2286f

Please sign in to comment.