99 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-ch5",62 srcs = [63 "mlir/Dialect.cpp",64 "mlir/LowerToAffineLoops.cpp",65 "mlir/MLIRGen.cpp",66 "mlir/ShapeInferencePass.cpp",67 "mlir/ToyCombine.cpp",68 "parser/AST.cpp",69 "toyc.cpp",70 ] + glob(["include/toy/*.h"]),71 includes = ["include/"],72 deps = [73 ":ToyCombineIncGen",74 ":ToyInterfacesIncGen",75 ":ToyOpsIncGen",76 "//llvm:Support",77 "//mlir:AffineDialect",78 "//mlir:AffineTransforms",79 "//mlir:AllPassesAndDialects",80 "//mlir:Analysis",81 "//mlir:ArithDialect",82 "//mlir:BytecodeReader",83 "//mlir:CallOpInterfaces",84 "//mlir:CastInterfaces",85 "//mlir:FuncDialect",86 "//mlir:FuncExtensions",87 "//mlir:FunctionInterfaces",88 "//mlir:IR",89 "//mlir:InliningUtils",90 "//mlir:MemRefDialect",91 "//mlir:Parser",92 "//mlir:Pass",93 "//mlir:SideEffectInterfaces",94 "//mlir:Support",95 "//mlir:TransformUtils",96 "//mlir:Transforms",97 ],98)99