From 7347c28def5637d3b44c6980383437672f43fdf5 Mon Sep 17 00:00:00 2001
From: Mogball <jeffniu22@gmail.com>
Date: Wed, 22 Dec 2021 20:00:00 +0000
Subject: [mlir] Add missing unit tests to BUILD.bazel

Several unit test folders were missing from the bazel build, including: Transforms, Conversion, Rewrite
---
 mlir/unittests/Analysis/CMakeLists.txt             |  2 +-
 .../mlir/unittests/BUILD.bazel                     | 43 ++++++++++++++++++++++
 2 files changed, 44 insertions(+), 1 deletion(-)

diff --git a/mlir/unittests/Analysis/CMakeLists.txt b/mlir/unittests/Analysis/CMakeLists.txt
index 3346426..b6340ec 100644
--- a/mlir/unittests/Analysis/CMakeLists.txt
+++ b/mlir/unittests/Analysis/CMakeLists.txt
@@ -7,7 +7,7 @@ add_mlir_unittest(MLIRAnalysisTests
 )
 
 target_link_libraries(MLIRAnalysisTests
-  PRIVATE 
+  PRIVATE
   MLIRLoopAnalysis
   MLIRParser
   )
diff --git a/utils/bazel/llvm-project-overlay/mlir/unittests/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/unittests/BUILD.bazel
index ff22e88..5ee06e7 100644
--- a/utils/bazel/llvm-project-overlay/mlir/unittests/BUILD.bazel
+++ b/utils/bazel/llvm-project-overlay/mlir/unittests/BUILD.bazel
@@ -76,6 +76,20 @@ cc_test(
 )
 
 cc_test(
+    name = "rewrite_tests",
+    size = "small",
+    srcs = glob([
+        "Rewrite/*.cpp",
+        "Rewrite/*.h",
+    ]),
+    deps = [
+        "//llvm:gtest_main",
+        "//mlir:IR",
+        "//mlir:Rewrite",
+    ],
+)
+
+cc_test(
     name = "dialect_tests",
     size = "small",
     srcs = glob([
@@ -244,6 +258,19 @@ cc_test(
 )
 
 cc_test(
+    name = "transforms_test",
+    size = "small",
+    srcs = glob([
+        "Transforms/*.cpp",
+        "Transforms/*.h",
+    ]),
+    deps = [
+        "//llvm:gtest_main",
+        "//mlir:TransformUtils",
+    ],
+)
+
+cc_test(
     name = "analysis_tests",
     size = "small",
     srcs = glob([
@@ -261,6 +288,22 @@ cc_test(
 )
 
 cc_test(
+    name = "conversion_tests",
+    size = "small",
+    srcs = glob([
+        "Conversion/*.cpp",
+        "Conversion/*.h",
+        "Conversion/*/*.cpp",
+        "Conversion/*/*.h",
+    ]),
+    deps = [
+        "//llvm:gtest_main",
+        "//mlir:ArithmeticDialect",
+        "//mlir:PDLToPDLInterp",
+    ],
+)
+
+cc_test(
     name = "execution_engine_tests",
     size = "small",
     srcs = glob([
-- 
cgit v1.1