From b0bd0b41ae15fcdb4e31d2f790ebb82cc16dbd67 Mon Sep 17 00:00:00 2001
From: Madoura <madouura@gmail.com>
Date: Thu, 3 Nov 2022 06:20:02 -0500
Subject: [PATCH] rocmlir: init at 5.3.0

---
 .../development/libraries/rocmlir/default.nix | 42 +++++++++++++++++++
 pkgs/top-level/all-packages.nix               |  2 +
 2 files changed, 44 insertions(+)
 create mode 100644 pkgs/development/libraries/rocmlir/default.nix

diff --git a/pkgs/development/libraries/rocmlir/default.nix b/pkgs/development/libraries/rocmlir/default.nix
new file mode 100644
index 000000000000..bf4234a21503
--- /dev/null
+++ b/pkgs/development/libraries/rocmlir/default.nix
@@ -0,0 +1,42 @@
+{ lib
+, stdenv
+, fetchFromGitHub
+, cmake
+, hip
+, python3
+}:
+
+stdenv.mkDerivation rec {
+  pname = "rocmlir";
+  rocmVersion = "5.3.1";
+  # For some reason they didn't add a tag for 5.3.1, should be compatible, change to rocmVersion later
+  version = "5.3.0";
+
+  src = fetchFromGitHub {
+    owner = "ROCmSoftwarePlatform";
+    repo = "rocMLIR";
+    rev = "rocm-${version}"; # change to rocmVersion later
+    hash = "sha256-s/5gAH5vh2tgATZemPP66juQFDg8BR2sipzX2Q6pOOQ=";
+  };
+
+  nativeBuildInputs = [
+    cmake
+    hip
+  ];
+
+  buildInputs = [
+    python3
+  ];
+
+  cmakeFlags = [
+    "-DBUILD_FAT_LIBMLIRMIOPEN=ON"
+  ];
+
+  meta = with lib; {
+    description = "MLIR-based convolution and GEMM kernel generator";
+    homepage = "https://github.com/ROCmSoftwarePlatform/rocMLIR";
+    license = with licenses; [ asl20 ];
+    maintainers = with maintainers; [ Madouura ];
+    broken = rocmVersion != hip.version;
+  };
+}
diff --git a/pkgs/top-level/all-packages.nix b/pkgs/top-level/all-packages.nix
index 7e5ed41963f2..c247264ae824 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 { };
 
+  rocmlir = callPackage ../development/libraries/rocmlir { };
+
   rtags = callPackage ../development/tools/rtags {
     inherit (darwin) apple_sdk;
   };