75 lines · plain
1# Description:2# MLIR Tutorial3 4load("@rules_cc//cc:defs.bzl", "cc_binary", "cc_library")5load("//mlir:tblgen.bzl", "gentbl_cc_library", "td_library")6 7licenses(["notice"])8 9package(default_visibility = ["//visibility:public"])10 11td_library(12 name = "TransformTdFiles",13 srcs = [14 "include/MyExtension.td",15 ],16 includes = ["include"],17 deps = [18 "//mlir:OpBaseTdFiles",19 "//mlir:SideEffectInterfacesTdFiles",20 "//mlir:TransformDialectTdFiles",21 ],22)23 24gentbl_cc_library(25 name = "TransformIncGen",26 tbl_outs = [27 (28 ["-gen-op-decls"],29 "include/MyExtension.h.inc",30 ),31 (32 ["-gen-op-defs"],33 "include/MyExtension.cpp.inc",34 ),35 ],36 tblgen = "//mlir:mlir-tblgen",37 td_file = "include/MyExtension.td",38 deps = [":TransformTdFiles"],39)40 41cc_library(42 name = "MyExtension",43 srcs = glob(["lib/*.cpp"]),44 hdrs = glob(["include/*.h"]),45 includes = ["include"],46 deps = [47 ":TransformIncGen",48 "//llvm:Support",49 "//mlir:FuncDialect",50 "//mlir:IR",51 "//mlir:SCFDialect",52 "//mlir:Support",53 "//mlir:TransformDialect",54 "//mlir:TransformDialectInterfaces",55 ],56)57 58cc_binary(59 name = "transform-opt-ch4",60 srcs = glob(["transform-opt/*.cpp"]),61 local_defines = ["MLIR_INCLUDE_TESTS"],62 deps = [63 ":MyExtension",64 "//mlir:AllExtensions",65 "//mlir:AllPassesAndDialects",66 "//mlir:IR",67 "//mlir:MlirOptLib",68 "//mlir:RegisterAllDialects",69 "//mlir:RegisterAllExtensions",70 "//mlir:TransformDialectTransforms",71 "//mlir:Transforms",72 "//mlir/test:TestTransformDialect",73 ],74)75