diff --git a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel --- a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel +++ b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel @@ -5939,6 +5939,7 @@ ":LinalgToSPIRV", ":LinalgToStandard", ":LinalgTransforms", + ":MLProgramDialect", ":MathDialect", ":MathToLLVM", ":MathToLibm", @@ -8114,6 +8115,77 @@ ], ) +##---------------------------------------------------------------------------## +# 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"],