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

   1 # This file is licensed under the Apache License v2.0 with LLVM Exceptions.
   2 # See https://llvm.org/LICENSE.txt for license information.
   3 # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   4
   5 load("//mlir:tblgen.bzl", "gentbl_cc_library")
   6
   7 package(
   8     default_visibility = ["//visibility:public"],
   9     features = ["layering_check"],
  10 )
  11
  12 licenses(["notice"])
  13
  14 cc_test(
  15     name = "debug_tests",
  16     size = "small",
  17     srcs = glob([
  18         "Debug/*.cpp",
  19     ]),
  20     deps = [
  21         "//llvm:Support",
  22         "//llvm:TestingSupport",
  23         "//mlir:Debug",
  24         "//mlir:IR",
  25         "//mlir:Support",
  26         "//third-party/unittest:gmock",
  27         "//third-party/unittest:gtest",
  28         "//third-party/unittest:gtest_main",
  29     ],
  30 )
  31
  32 cc_test(
  33     name = "ir_tests",
  34     size = "small",
  35     srcs = glob([
  36         "IR/*.cpp",
  37     ]),
  38     deps = [
  39         "//llvm:Support",
  40         "//mlir:BytecodeReader",
  41         "//mlir:CallOpInterfaces",
  42         "//mlir:FunctionInterfaces",
  43         "//mlir:IR",
  44         "//mlir:Parser",
  45         "//mlir:Support",
  46         "//mlir/test:TestDialect",
  47         "//third-party/unittest:gmock",
  48         "//third-party/unittest:gtest",
  49         "//third-party/unittest:gtest_main",
  50     ],
  51 )
  52
  53 cc_test(
  54     name = "interface_tests",
  55     size = "small",
  56     srcs = glob([
  57         "Interfaces/*.cpp",
  58     ]),
  59     deps = [
  60         "//llvm:Support",
  61         "//llvm:TestingSupport",
  62         "//mlir:ArithDialect",
  63         "//mlir:BytecodeReader",
  64         "//mlir:ControlFlowInterfaces",
  65         "//mlir:DLTIDialect",
  66         "//mlir:DataLayoutInterfaces",
  67         "//mlir:FuncDialect",
  68         "//mlir:IR",
  69         "//mlir:InferIntRangeInterface",
  70         "//mlir:InferTypeOpInterface",
  71         "//mlir:Parser",
  72         "//third-party/unittest:gtest",
  73         "//third-party/unittest:gtest_main",
  74     ],
  75 )
  76
  77 cc_test(
  78     name = "support_tests",
  79     size = "small",
  80     srcs = glob([
  81         "Support/*.cpp",
  82     ]),
  83     deps = [
  84         "//llvm:Support",
  85         "//llvm:TestingSupport",
  86         "//mlir:Support",
  87         "//third-party/unittest:gmock",
  88         "//third-party/unittest:gtest_main",
  89     ],
  90 )
  91
  92 cc_test(
  93     name = "pass_tests",
  94     size = "small",
  95     srcs = glob([
  96         "Pass/*.cpp",
  97     ]),
  98     deps = [
  99         "//llvm:Support",
 100         "//llvm:TestingSupport",
 101         "//mlir:Analysis",
 102         "//mlir:Debug",
 103         "//mlir:FuncDialect",
 104         "//mlir:IR",
 105         "//mlir:Pass",
 106         "//third-party/unittest:gtest",
 107         "//third-party/unittest:gtest_main",
 108     ],
 109 )
 110
 111 cc_test(
 112     name = "rewrite_tests",
 113     size = "small",
 114     srcs = glob([
 115         "Rewrite/*.cpp",
 116     ]),
 117     deps = [
 118         "//mlir:IR",
 119         "//mlir:Rewrite",
 120         "//mlir:SideEffectInterfaces",
 121         "//third-party/unittest:gtest",
 122         "//third-party/unittest:gtest_main",
 123     ],
 124 )
 125
 126 cc_test(
 127     name = "dialect_tests",
 128     size = "small",
 129     srcs = glob([
 130         "Dialect/*.cpp",
 131     ]),
 132     deps = [
 133         "//llvm:Support",
 134         "//mlir:Dialect",
 135         "//third-party/unittest:gmock",
 136         "//third-party/unittest:gtest_main",
 137     ],
 138 )
 139
 140 cc_test(
 141     name = "amdgpu_tests",
 142     size = "small",
 143     srcs = glob([
 144         "Dialect/AMDGPU/*.cpp",
 145     ]),
 146     deps = [
 147         "//mlir:AMDGPUUtils",
 148         "//third-party/unittest:gtest",
 149         "//third-party/unittest:gtest_main",
 150     ],
 151 )
 152
 153 cc_test(
 154     name = "memref_tests",
 155     size = "small",
 156     srcs = glob([
 157         "Dialect/MemRef/*.cpp",
 158     ]),
 159     deps = [
 160         "//llvm:TestingSupport",
 161         "//mlir:IR",
 162         "//mlir:MemRefDialect",
 163         "//mlir:SideEffectInterfaces",
 164         "//third-party/unittest:gtest",
 165         "//third-party/unittest:gtest_main",
 166     ],
 167 )
 168
 169 cc_test(
 170     name = "scf_tests",
 171     size = "small",
 172     srcs = glob([
 173         "Dialect/SCF/*.cpp",
 174     ]),
 175     deps = [
 176         "//mlir:ArithDialect",
 177         "//mlir:FuncDialect",
 178         "//mlir:IR",
 179         "//mlir:Parser",
 180         "//mlir:SCFDialect",
 181         "//mlir:SideEffectInterfaces",
 182         "//third-party/unittest:gtest",
 183         "//third-party/unittest:gtest_main",
 184     ],
 185 )
 186
 187 cc_test(
 188     name = "sparse_tensor_tests",
 189     size = "small",
 190     srcs = glob([
 191         "Dialect/SparseTensor/*.cpp",
 192     ]),
 193     deps = [
 194         "//llvm:Support",
 195         "//llvm:TestingSupport",
 196         "//mlir:SideEffectInterfaces",
 197         "//mlir:SparseTensorUtils",
 198         "//third-party/unittest:gmock",
 199         "//third-party/unittest:gtest",
 200         "//third-party/unittest:gtest_main",
 201     ],
 202 )
 203
 204 cc_test(
 205     name = "spirv_tests",
 206     size = "small",
 207     srcs = glob([
 208         "Dialect/SPIRV/*.cpp",
 209     ]),
 210     deps = [
 211         "//llvm:Support",
 212         "//llvm:TestingSupport",
 213         "//mlir:IR",
 214         "//mlir:SPIRVBinaryUtils",
 215         "//mlir:SPIRVDeserialization",
 216         "//mlir:SPIRVDialect",
 217         "//mlir:SPIRVSerialization",
 218         "//third-party/unittest:gmock",
 219         "//third-party/unittest:gtest_main",
 220     ],
 221 )
 222
 223 cc_test(
 224     name = "transform_dialect_tests",
 225     size = "small",
 226     srcs = glob([
 227         "Dialect/Transform/*.cpp",
 228     ]),
 229     deps = [
 230         "//llvm:Support",
 231         "//llvm:TestingSupport",
 232         "//mlir:BytecodeReader",
 233         "//mlir:FuncDialect",
 234         "//mlir:IR",
 235         "//mlir:MlirOptLib",
 236         "//mlir:Parser",
 237         "//mlir:Pass",
 238         "//mlir:Support",
 239         "//mlir:TransformDebugExtension",
 240         "//mlir:TransformDialect",
 241         "//mlir:TransformDialectTransforms",
 242         "//mlir/test:TestTransformDialect",
 243         "//third-party/unittest:gtest",
 244         "//third-party/unittest:gtest_main",
 245     ],
 246 )
 247
 248 cc_test(
 249     name = "dialect_utils_tests",
 250     size = "small",
 251     srcs = glob([
 252         "Dialect/Utils/*.cpp",
 253     ]),
 254     deps = [
 255         "//llvm:Support",
 256         "//llvm:TestingSupport",
 257         "//mlir:DialectUtils",
 258         "//mlir:IR",
 259         "//third-party/unittest:gmock",
 260         "//third-party/unittest:gtest",
 261         "//third-party/unittest:gtest_main",
 262     ],
 263 )
 264
 265 gentbl_cc_library(
 266     name = "EnumsIncGen",
 267     tbl_outs = [
 268         (
 269             ["-gen-enum-decls"],
 270             "TableGen/EnumsGenTest.h.inc",
 271         ),
 272         (
 273             ["-gen-enum-defs"],
 274             "TableGen/EnumsGenTest.cpp.inc",
 275         ),
 276     ],
 277     tblgen = "//mlir:mlir-tblgen",
 278     td_file = "TableGen/enums.td",
 279     deps = [
 280         "//mlir:OpBaseTdFiles",
 281     ],
 282 )
 283
 284 gentbl_cc_library(
 285     name = "PassIncGen",
 286     tbl_outs = [
 287         (
 288             ["-gen-pass-decls"],
 289             "TableGen/PassGenTest.h.inc",
 290         ),
 291     ],
 292     tblgen = "//mlir:mlir-tblgen",
 293     td_file = "TableGen/passes.td",
 294     deps = [
 295         "//mlir:PassBaseTdFiles",
 296         "//mlir:RewritePassBaseTdFiles",
 297     ],
 298 )
 299
 300 cc_test(
 301     name = "tablegen_tests",
 302     size = "small",
 303     srcs = glob([
 304         "TableGen/*.cpp",
 305     ]) + [
 306         "TableGen/EnumsGenTest.cpp.inc",
 307         "TableGen/EnumsGenTest.h.inc",
 308     ],
 309     includes = ["TableGen/"],
 310     deps = [
 311         ":EnumsIncGen",
 312         ":PassIncGen",
 313         "//llvm:Support",
 314         "//llvm:TestingSupport",
 315         "//mlir:IR",
 316         "//mlir:Pass",
 317         "//mlir:Support",
 318         "//mlir:TableGen",
 319         "//mlir/test:TestDialect",
 320         "//third-party/unittest:gmock",
 321         "//third-party/unittest:gtest_main",
 322     ],
 323 )
 324
 325 cc_test(
 326     name = "transforms_test",
 327     size = "small",
 328     srcs = glob([
 329         "Transforms/*.cpp",
 330     ]),
 331     deps = [
 332         "//mlir:AffineAnalysis",
 333         "//mlir:BytecodeReader",
 334         "//mlir:IR",
 335         "//mlir:Parser",
 336         "//mlir:Pass",
 337         "//mlir:SideEffectInterfaces",
 338         "//mlir:TransformUtils",
 339         "//mlir:Transforms",
 340         "//third-party/unittest:gtest",
 341         "//third-party/unittest:gtest_main",
 342     ],
 343 )
 344
 345 cc_test(
 346     name = "analysis_tests",
 347     size = "small",
 348     srcs = glob([
 349         "Analysis/*/*.cpp",
 350         "Analysis/*/*.h",
 351     ]),
 352     deps = [
 353         "//llvm:TestingSupport",
 354         "//mlir:AffineAnalysis",
 355         "//mlir:Analysis",
 356         "//mlir:AsmParser",
 357         "//mlir:IR",
 358         "//mlir:Parser",
 359         "//mlir:Support",
 360         "//third-party/unittest:gmock",
 361         "//third-party/unittest:gtest",
 362         "//third-party/unittest:gtest_main",
 363     ],
 364 )
 365
 366 cc_test(
 367     name = "bytecode_tests",
 368     size = "small",
 369     srcs = glob([
 370         "Bytecode/*.cpp",
 371     ]),
 372     deps = [
 373         "//llvm:Support",
 374         "//mlir:BytecodeReader",
 375         "//mlir:BytecodeWriter",
 376         "//mlir:IR",
 377         "//mlir:Parser",
 378         "//mlir:SideEffectInterfaces",
 379         "//third-party/unittest:gmock",
 380         "//third-party/unittest:gtest",
 381         "//third-party/unittest:gtest_main",
 382     ],
 383 )
 384
 385 cc_test(
 386     name = "conversion_tests",
 387     size = "small",
 388     srcs = glob([
 389         "Conversion/*/*.cpp",
 390     ]),
 391     deps = [
 392         "//mlir:ArithDialect",
 393         "//mlir:IR",
 394         "//mlir:PDLToPDLInterp",
 395         "//third-party/unittest:gtest",
 396         "//third-party/unittest:gtest_main",
 397     ],
 398 )
 399
 400 cc_test(
 401     name = "execution_engine_tests",
 402     size = "small",
 403     srcs = glob([
 404         "ExecutionEngine/*.cpp",
 405     ]),
 406     tags = [
 407         # MSAN does not work with JIT.
 408         "nomsan",
 409     ],
 410     deps = [
 411         "//llvm:Support",
 412         "//llvm:TestingSupport",
 413         "//mlir:AllPassesAndDialects",
 414         "//mlir:Analysis",
 415         "//mlir:ArithToLLVM",
 416         "//mlir:BuiltinToLLVMIRTranslation",
 417         "//mlir:BytecodeReader",
 418         "//mlir:ExecutionEngine",
 419         "//mlir:FuncDialect",
 420         "//mlir:FuncToLLVM",
 421         "//mlir:IR",
 422         "//mlir:LLVMToLLVMIRTranslation",
 423         "//mlir:LinalgTransforms",
 424         "//mlir:MemRefToLLVM",
 425         "//mlir:Parser",
 426         "//mlir:Pass",
 427         "//mlir:ReconcileUnrealizedCasts",
 428         "//mlir:ToLLVMIRTranslation",
 429         "//mlir:VectorToLLVM",
 430         "//mlir:VectorToSCF",
 431         "//mlir:mlir_c_runner_utils",
 432         "//mlir:mlir_runner_utils",
 433         "//third-party/unittest:gmock",
 434         "//third-party/unittest:gtest_main",
 435     ],
 436 )