pyrox.dev
lol
rocmlir: init at 5.3.0
+42
pkgs/development/libraries/rocmlir/default.nix
+42
pkgs/development/libraries/rocmlir/default.nix
···
1
+
{ lib
2
+
, stdenv
3
+
, fetchFromGitHub
4
+
, cmake
5
+
, hip
6
+
, python3
7
+
}:
8
+
9
+
stdenv.mkDerivation rec {
10
+
pname = "rocmlir";
11
+
rocmVersion = "5.3.1";
12
+
# For some reason they didn't add a tag for 5.3.1, should be compatible, change to rocmVersion later
13
+
version = "5.3.0";
14
+
15
+
src = fetchFromGitHub {
16
+
owner = "ROCmSoftwarePlatform";
17
+
repo = "rocMLIR";
18
+
rev = "rocm-${version}"; # change to rocmVersion later
19
+
hash = "sha256-s/5gAH5vh2tgATZemPP66juQFDg8BR2sipzX2Q6pOOQ=";
20
+
};
21
+
22
+
nativeBuildInputs = [
23
+
cmake
24
+
hip
25
+
];
26
+
27
+
buildInputs = [
28
+
python3
29
+
];
30
+
31
+
cmakeFlags = [
32
+
"-DBUILD_FAT_LIBMLIRMIOPEN=ON"
33
+
];
34
+
35
+
meta = with lib; {
36
+
description = "MLIR-based convolution and GEMM kernel generator";
37
+
homepage = "https://github.com/ROCmSoftwarePlatform/rocMLIR";
38
+
license = with licenses; [ asl20 ];
39
+
maintainers = with maintainers; [ Madouura ];
40
+
broken = rocmVersion != hip.version;
41
+
};
42
+
}
+2
pkgs/top-level/all-packages.nix
+2
pkgs/top-level/all-packages.nix