[mlir] Add mlir/unittests/BUILD.bazel

Unit tests are not getting built as part of bazel runs.

Reviewed By: mehdi_amini, GMNGeoffrey

Differential Revision: https://reviews.llvm.org/D116046

utils/bazel/llvm-project-overlay/mlir/unittests/BUILD.bazel

@@ -0,0 +1,270 @@
+# This file is licensed under the Apache License v2.0 with LLVM Exceptions.
+# See https://llvm.org/LICENSE.txt for license information.
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
+load("//mlir:tblgen.bzl", "gentbl_cc_library")
+
+package(
+    default_visibility = ["//visibility:public"],
+    licenses = ["notice"],
+)
+
+cc_test(
+    name = "ir_tests",
+    size = "small",
+    srcs = glob([
+        "IR/*.cpp",
+        "IR/*.h",
+    ]),
+    deps = [
+        "//llvm:TestingSupport",
+        "//llvm:gtest_main",
+        "//mlir:IR",
+        "//mlir/test:TestDialect",
+    ],
+)
+
+cc_test(
+    name = "interface_tests",
+    size = "small",
+    srcs = glob([
+        "Interfaces/*.cpp",
+        "Interfaces/*.h",
+    ]),
+    deps = [
+        "//llvm:TestingSupport",
+        "//llvm:gtest_main",
+        "//mlir:ControlFlowInterfaces",
+        "//mlir:DLTIDialect",
+        "//mlir:DataLayoutInterfaces",
+        "//mlir:IR",
+        "//mlir:InferTypeOpInterface",
+        "//mlir:Parser",
+        "//mlir:StandardOps",
+    ],
+)
+
+cc_test(
+    name = "support_tests",
+    size = "small",
+    srcs = glob([
+        "Support/*.cpp",
+        "Support/*.h",
+    ]),
+    deps = [
+        "//llvm:Support",
+        "//llvm:TestingSupport",
+        "//llvm:gtest_main",
+        "//mlir:Support",
+    ],
+)
+
+cc_test(
+    name = "pass_tests",
+    size = "small",
+    srcs = glob([
+        "Pass/*.cpp",
+        "Pass/*.h",
+    ]),
+    deps = [
+        "//llvm:TestingSupport",
+        "//llvm:gtest_main",
+        "//mlir:Analysis",
+        "//mlir:IR",
+        "//mlir:Pass",
+    ],
+)
+
+cc_test(
+    name = "dialect_tests",
+    size = "small",
+    srcs = glob([
+        "Dialect/*.cpp",
+        "Dialect/*.h",
+    ]),
+    deps = [
+        "//llvm:TestingSupport",
+        "//llvm:gtest_main",
+        "//mlir:Dialect",
+    ],
+)
+
+cc_test(
+    name = "quantops_tests",
+    size = "small",
+    srcs = glob([
+        "Dialect/Quant/*.cpp",
+        "Dialect/Quant/*.h",
+    ]),
+    deps = [
+        "//llvm:TestingSupport",
+        "//llvm:gtest_main",
+        "//mlir:QuantOps",
+        "//mlir:Transforms",
+    ],
+)
+
+cc_test(
+    name = "sparse_tensor_tests",
+    size = "small",
+    srcs = glob([
+        "Dialect/SparseTensor/*.cpp",
+        "Dialect/SparseTensor/*.h",
+    ]),
+    deps = [
+        "//llvm:TestingSupport",
+        "//llvm:gtest_main",
+        "//mlir:SparseTensorUtils",
+    ],
+)
+
+cc_test(
+    name = "spirv_tests",
+    size = "small",
+    srcs = glob([
+        "Dialect/SPIRV/*.cpp",
+        "Dialect/SPIRV/*.h",
+    ]),
+    deps = [
+        "//llvm:TestingSupport",
+        "//llvm:gtest_main",
+        "//mlir:SPIRVDeserialization",
+        "//mlir:SPIRVDialect",
+        "//mlir:SPIRVSerialization",
+    ],
+)
+
+cc_test(
+    name = "dialect_utils_tests",
+    size = "small",
+    srcs = glob([
+        "Dialect/Utils/*.cpp",
+        "Dialect/Utils/*.h",
+    ]),
+    deps = [
+        "//llvm:TestingSupport",
+        "//llvm:gtest_main",
+        "//mlir:DialectUtils",
+    ],
+)
+
+gentbl_cc_library(
+    name = "EnumsIncGen",
+    tbl_outs = [
+        (
+            ["-gen-enum-decls"],
+            "TableGen/EnumsGenTest.h.inc",
+        ),
+        (
+            ["-gen-enum-defs"],
+            "TableGen/EnumsGenTest.cpp.inc",
+        ),
+    ],
+    tblgen = "//mlir:mlir-tblgen",
+    td_file = "TableGen/enums.td",
+    deps = [
+        "//mlir:OpBaseTdFiles",
+    ],
+)
+
+gentbl_cc_library(
+    name = "PassIncGen",
+    tbl_outs = [
+        (
+            ["-gen-pass-decls"],
+            "TableGen/PassGenTest.h.inc",
+        ),
+    ],
+    tblgen = "//mlir:mlir-tblgen",
+    td_file = "TableGen/passes.td",
+    deps = [
+        "//mlir:PassBaseTdFiles",
+        "//mlir:RewritePassBaseTdFiles",
+    ],
+)
+
+gentbl_cc_library(
+    name = "StructsIncGen",
+    tbl_outs = [
+        (
+            ["-gen-struct-attr-decls"],
+            "TableGen/StructAttrGenTest.h.inc",
+        ),
+        (
+            ["-gen-struct-attr-defs"],
+            "TableGen/StructAttrGenTest.cpp.inc",
+        ),
+    ],
+    tblgen = "//mlir:mlir-tblgen",
+    td_file = "TableGen/structs.td",
+    deps = [
+        "//mlir:OpBaseTdFiles",
+    ],
+)
+
+cc_test(
+    name = "tablegen_tests",
+    size = "small",
+    srcs = glob([
+        "TableGen/*.cpp",
+        "TableGen/*.h",
+    ]) + [
+        "TableGen/EnumsGenTest.cpp.inc",
+        "TableGen/EnumsGenTest.h.inc",
+        "TableGen/StructAttrGenTest.cpp.inc",
+        "TableGen/StructAttrGenTest.h.inc",
+    ],
+    includes = ["TableGen/"],
+    deps = [
+        ":EnumsIncGen",
+        ":PassIncGen",
+        ":StructsIncGen",
+        "//llvm:Support",
+        "//llvm:TestingSupport",
+        "//llvm:gtest_main",
+        "//mlir:IR",
+        "//mlir:TableGen",
+        "//mlir/test:TestDialect",
+    ],
+)
+
+cc_test(
+    name = "analysis_tests",
+    size = "small",
+    srcs = glob([
+        "Analysis/*.cpp",
+        "Analysis/*.h",
+        "Analysis/*/*.cpp",
+        "Analysis/*/*.h",
+    ]),
+    deps = [
+        "//llvm:TestingSupport",
+        "//llvm:gtest_main",
+        "//mlir:Analysis",
+        "//mlir:IR",
+    ],
+)
+
+cc_test(
+    name = "execution_engine_tests",
+    size = "small",
+    srcs = glob([
+        "ExecutionEngine/*.cpp",
+    ]),
+    tags = [
+        # MSAN does not work with JIT.
+        "nomsan",
+    ],
+    deps = [
+        "//llvm:TestingSupport",
+        "//llvm:gtest_main",
+        "//mlir:AllPassesAndDialects",
+        "//mlir:Analysis",
+        "//mlir:ExecutionEngine",
+        "//mlir:IR",
+        "//mlir:LinalgToLLVM",
+        "//mlir:MemRefToLLVM",
+        "//mlir:mlir_c_runner_utils",
+        "//mlir:mlir_runner_utils",
+    ],
+)