nixpkgs/pkgs/development/libraries/rocmlir/default.nix

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

58 lines
1.3 KiB
Nix
Raw Normal View History

2022-11-03 11:20:02 +00:00
{ lib
, stdenv
, fetchFromGitHub
, writeScript
2022-11-03 11:20:02 +00:00
, cmake
, clang
, git
, libxml2
, libedit
2022-11-03 11:20:02 +00:00
, python3
}:
stdenv.mkDerivation (finalAttrs: {
2022-11-03 11:20:02 +00:00
pname = "rocmlir";
version = "5.4.0";
2022-11-03 11:20:02 +00:00
src = fetchFromGitHub {
owner = "ROCmSoftwarePlatform";
repo = "rocMLIR";
rev = "rocm-${finalAttrs.version}";
hash = "sha256-MokE7Ej8mLHTQeLYvKr7PPlsNG6ul91fqfXDlGu5JpI=";
2022-11-03 11:20:02 +00:00
};
nativeBuildInputs = [
cmake
clang
2022-11-03 11:20:02 +00:00
];
buildInputs = [
git
libxml2
libedit
2022-11-03 11:20:02 +00:00
python3
];
cmakeFlags = [
"-DCMAKE_C_COMPILER=clang"
"-DCMAKE_CXX_COMPILER=clang++"
"-DBUILD_FAT_LIBROCKCOMPILER=ON"
2022-11-03 11:20:02 +00:00
];
passthru.updateScript = writeScript "update.sh" ''
#!/usr/bin/env nix-shell
#!nix-shell -i bash -p curl jq common-updater-scripts
version="$(curl ''${GITHUB_TOKEN:+"-u \":$GITHUB_TOKEN\""} \
-sL "https://api.github.com/repos/ROCmSoftwarePlatform/rocMLIR/tags?per_page=2" | jq '.[1].name | split("-") | .[1]' --raw-output)"
update-source-version rocmlir "$version" --ignore-same-hash
'';
2022-11-03 11:20:02 +00:00
meta = with lib; {
description = "MLIR-based convolution and GEMM kernel generator";
homepage = "https://github.com/ROCmSoftwarePlatform/rocMLIR";
license = with licenses; [ asl20 ];
maintainers = teams.rocm.members;
broken = finalAttrs.version != clang.version;
2022-11-03 11:20:02 +00:00
};
})