diff options
| author | Stella Laurenzo <stellaraccident@gmail.com> | 2022-04-14 06:16:04 +0300 |
|---|---|---|
| committer | Stella Laurenzo <stellaraccident@gmail.com> | 2022-04-14 07:38:14 +0300 |
| commit | 61352a580a1f8e5818a6e5445517058d959bb86f (patch) | |
| tree | f1aeb62845e97c875db82a990a4105d10909c25d /utils | |
| parent | 836e610d9332772ad4511e86ef126a200f1ab281 (diff) | |
[mlir] Introduce ml_program dialect.
Differential Revision: https://reviews.llvm.org/D120203
Diffstat (limited to 'utils')
| -rw-r--r-- | utils/bazel/llvm-project-overlay/mlir/BUILD.bazel | 72 |
1 files changed, 72 insertions, 0 deletions
diff --git a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel index ba87c8e49960..85d94212fd72 100644 --- a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel +++ b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel @@ -5939,6 +5939,7 @@ cc_library( ":LinalgToSPIRV", ":LinalgToStandard", ":LinalgTransforms", + ":MLProgramDialect", ":MathDialect", ":MathToLLVM", ":MathToLibm", @@ -8114,6 +8115,77 @@ cc_library( ], ) +##---------------------------------------------------------------------------## +# MLProgram dialect +##---------------------------------------------------------------------------## + +td_library( + name = "MLProgramOpsTdFiles", + srcs = [ + "include/mlir/Dialect/MLProgram/IR/MLProgramBase.td", + "include/mlir/Dialect/MLProgram/IR/MLProgramOps.td", + ], + includes = ["include"], + deps = [ + ":CallInterfacesTdFiles", + ":ControlFlowInterfacesTdFiles", + ":FunctionInterfacesTdFiles", + ":OpBaseTdFiles", + ":RegionKindInterfaceIncGen", + ":SideEffectInterfacesTdFiles", + ], +) + +gentbl_cc_library( + name = "MLProgramOpsIncGen", + strip_include_prefix = "include", + tbl_outs = [ + ( + ["-gen-op-decls"], + "include/mlir/Dialect/MLProgram/IR/MLProgramOps.h.inc", + ), + ( + ["-gen-op-defs"], + "include/mlir/Dialect/MLProgram/IR/MLProgramOps.cpp.inc", + ), + ( + ["-gen-dialect-decls"], + "include/mlir/Dialect/MLProgram/IR/MLProgramOpsDialect.h.inc", + ), + ( + ["-gen-dialect-defs"], + "include/mlir/Dialect/MLProgram/IR/MLProgramOpsDialect.cpp.inc", + ), + ], + tblgen = ":mlir-tblgen", + td_file = "include/mlir/Dialect/MLProgram/IR/MLProgramOps.td", + deps = [":MLProgramOpsTdFiles"], +) + +cc_library( + name = "MLProgramDialect", + srcs = glob([ + "lib/Dialect/MLProgram/IR/*.cpp", + "lib/Dialect/MLProgram/IR/*.h", + ]), + hdrs = glob([ + "include/mlir/Dialect/MLProgram/IR/*.h", + ]), + includes = ["include"], + deps = [ + ":ControlFlowInterfaces", + ":IR", + ":MLProgramOpsIncGen", + ":Pass", + ":Support", + "//llvm:Support", + ], +) + +##---------------------------------------------------------------------------## +# Allocation interfaces +##---------------------------------------------------------------------------## + td_library( name = "AllocationOpInterfaceTdFiles", srcs = ["include/mlir/Dialect/Bufferization/IR/AllocationOpInterface.td"], |