pyrox.dev
lol
1{ lib
2, stdenv
3, fetchFromGitHub
4, cmake
5, hip
6, python3
7}:
8
9stdenv.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}