From bb2286f2d83bef8057c2e8a4fb3e4ade44dc128f Mon Sep 17 00:00:00 2001 From: Madoura Date: Tue, 25 Oct 2022 06:44:37 -0500 Subject: [PATCH] tensile: init at 4.34.0-5.3.1 --- .../development/libraries/tensile/default.nix | 34 +++++++++++++++++++ pkgs/top-level/all-packages.nix | 2 ++ 2 files changed, 36 insertions(+) create mode 100644 pkgs/development/libraries/tensile/default.nix diff --git a/pkgs/development/libraries/tensile/default.nix b/pkgs/development/libraries/tensile/default.nix new file mode 100644 index 0000000000000..d957ea7494419 --- /dev/null +++ b/pkgs/development/libraries/tensile/default.nix @@ -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 ]; + }; +} diff --git a/pkgs/top-level/all-packages.nix b/pkgs/top-level/all-packages.nix index 7e5ed41963f27..0e5725806253d 100644 --- a/pkgs/top-level/all-packages.nix +++ b/pkgs/top-level/all-packages.nix @@ -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; };