[bazel] Port cbd4750 (mlir/lib/Dialect/MLProgram)

llvm · Sep 19, 2023 · 7db4a6f · 7db4a6f
1 parent dc810e8
commit 7db4a6f
Showing 1 changed file with 37 additions and 2 deletions.
diff --git 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
 ##---------------------------------------------------------------------------##