91 lines · plain
1# Description:2# MLIR Tutorial3 4load("@rules_cc//cc:defs.bzl", "cc_binary")5load("//mlir:tblgen.bzl", "gentbl_cc_library", "td_library")6 7licenses(["notice"])8 9package(default_visibility = ["//visibility:public"])10 11td_library(12 name = "ToyOpsTdFiles",13 srcs = [14 "include/toy/Ops.td",15 "include/toy/ShapeInferenceInterface.td",16 ],17 includes = ["include"],18 deps = [19 "//mlir:CallInterfacesTdFiles",20 "//mlir:CastInterfacesTdFiles",21 "//mlir:FunctionInterfacesTdFiles",22 "//mlir:OpBaseTdFiles",23 "//mlir:SideEffectInterfacesTdFiles",24 ],25)26 27gentbl_cc_library(28 name = "ToyInterfacesIncGen",29 tbl_outs = {30 "include/toy/ShapeInferenceOpInterfaces.h.inc": ["-gen-op-interface-decls"],31 "include/toy/ShapeInferenceOpInterfaces.cpp.inc": ["-gen-op-interface-defs"],32 },33 tblgen = "//mlir:mlir-tblgen",34 td_file = "include/toy/ShapeInferenceInterface.td",35 deps = [":ToyOpsTdFiles"],36)37 38gentbl_cc_library(39 name = "ToyOpsIncGen",40 tbl_outs = {41 "include/toy/Ops.h.inc": ["-gen-op-decls"],42 "include/toy/Ops.cpp.inc": ["-gen-op-defs"],43 "include/toy/Dialect.h.inc": ["-gen-dialect-decls"],44 "include/toy/Dialect.cpp.inc": ["-gen-dialect-defs"],45 },46 tblgen = "//mlir:mlir-tblgen",47 td_file = "include/toy/Ops.td",48 deps = [":ToyOpsTdFiles"],49)50 51gentbl_cc_library(52 name = "ToyCombineIncGen",53 strip_include_prefix = "mlir",54 tbl_outs = {"mlir/ToyCombine.inc": ["-gen-rewriters"]},55 tblgen = "//mlir:mlir-tblgen",56 td_file = "mlir/ToyCombine.td",57 deps = [":ToyOpsTdFiles"],58)59 60cc_binary(61 name = "toyc-ch4",62 srcs = [63 "mlir/Dialect.cpp",64 "mlir/MLIRGen.cpp",65 "mlir/ShapeInferencePass.cpp",66 "mlir/ToyCombine.cpp",67 "parser/AST.cpp",68 "toyc.cpp",69 ] + glob(["include/toy/*.h"]),70 includes = ["include/"],71 deps = [72 ":ToyCombineIncGen",73 ":ToyInterfacesIncGen",74 ":ToyOpsIncGen",75 "//llvm:Support",76 "//mlir:Analysis",77 "//mlir:BytecodeReader",78 "//mlir:CallOpInterfaces",79 "//mlir:CastInterfaces",80 "//mlir:FunctionInterfaces",81 "//mlir:IR",82 "//mlir:InliningUtils",83 "//mlir:Parser",84 "//mlir:Pass",85 "//mlir:SideEffectInterfaces",86 "//mlir:Support",87 "//mlir:TransformUtils",88 "//mlir:Transforms",89 ],90)91