From 7db4a6f278a8b983a9ce86fa32f61ca9f9be5156 Mon Sep 17 00:00:00 2001
From: Fangrui Song <i@maskray.me>
Date: Mon, 18 Sep 2023 18:47:21 -0700
Subject: [PATCH] [bazel] Port cbd475040f8952cfc55b9e13dd5ce6c4f6434cd3
 (mlir/lib/Dialect/MLProgram)

---
 .../llvm-project-overlay/mlir/BUILD.bazel     | 39 ++++++++++++++++++-
 1 file changed, 37 insertions(+), 2 deletions(-)

diff --git a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
index 5d1574162aa690..2992bc3832ceb6 100644
--- a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
+++ b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel
@@ -8290,6 +8290,7 @@ cc_library(
         ":LinalgTransformOps",
         ":LinalgTransforms",
         ":MLProgramDialect",
+        ":MLProgramTransforms",
         ":MathDialect",
         ":MathToFuncs",
         ":MathToLLVM",
@@ -11871,6 +11872,22 @@ gentbl_cc_library(
     deps = [":MLProgramOpsTdFiles"],
 )
 
+gentbl_cc_library(
+    name = "MLProgramPassIncGen",
+    tbl_outs = [
+        (
+            [
+                "-gen-pass-decls",
+                "-name=MLProgram",
+            ],
+            "include/mlir/Dialect/MLProgram/Transforms/Passes.h.inc",
+        ),
+    ],
+    tblgen = ":mlir-tblgen",
+    td_file = "include/mlir/Dialect/MLProgram/Transforms/Passes.td",
+    deps = [":PassBaseTdFiles"],
+)
+
 gentbl_cc_library(
     name = "MLProgramTypesIncGen",
     tbl_outs = [
@@ -11892,10 +11909,9 @@ cc_library(
     name = "MLProgramDialect",
     srcs = glob([
         "lib/Dialect/MLProgram/IR/*.cpp",
-        "lib/Dialect/MLProgram/IR/*.h",
     ]),
     hdrs = glob([
-        "include/mlir/Dialect/MLProgram/IR/*.h",
+        "include/mlir/Dialect/MLProgram/**/*.h",
     ]),
     includes = ["include"],
     deps = [
@@ -11911,6 +11927,25 @@ cc_library(
     ],
 )
 
+cc_library(
+    name = "MLProgramTransforms",
+    srcs = glob([
+        "lib/Dialect/MLProgram/Transforms/*.cpp",
+    ]),
+    hdrs = glob([
+        "include/mlir/Dialect/MLProgram/Transforms/*.h",
+    ]),
+    includes = ["include"],
+    deps = [
+        ":MLProgramDialect",
+        ":FuncDialect",
+        ":TransformUtils",
+        ":IR",
+        ":MLProgramPassIncGen",
+        ":Pass",
+    ],
+)
+
 ##---------------------------------------------------------------------------##
 # Allocation interfaces
 ##---------------------------------------------------------------------------##