diff --git a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel index 4681a02ad50ba..089f284d3182a 100644 --- a/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel +++ b/utils/bazel/llvm-project-overlay/mlir/BUILD.bazel @@ -1211,6 +1211,20 @@ cc_binary( ], ) +cc_binary( + name = "_mlirDialectsIRDL.so", + srcs = ["lib/Bindings/Python/DialectIRDL.cpp"], + copts = PYBIND11_COPTS, + features = PYBIND11_FEATURES, + linkshared = 1, + linkstatic = 0, + deps = [ + ":CAPIIR", + ":MLIRBindingsPythonNanobindHeadersAndDeps", + "@nanobind", + ], +) + cc_binary( name = "_mlirDialectsLinalg.so", srcs = ["lib/Bindings/Python/DialectLinalg.cpp"], diff --git a/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel index 016794d30b349..a6e8c515f0874 100644 --- a/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel +++ b/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel @@ -537,6 +537,38 @@ filegroup( ], ) +##---------------------------------------------------------------------------## +# IRDL dialect. +##---------------------------------------------------------------------------## + +gentbl_filegroup( + name = "IRDLOpsPyGen", + tbl_outs = { + "mlir/dialects/_irdl_enum_gen.py": [ + "-gen-python-enum-bindings", + "-bind-dialect=irdl", + ], + "mlir/dialects/_irdl_ops_gen.py": [ + "-gen-python-op-bindings", + "-bind-dialect=irdl", + ], + }, + tblgen = "//mlir:mlir-tblgen", + td_file = "mlir/dialects/IRDLOps.td", + deps = [ + "//mlir:IRDLTdFiles", + "//mlir:OpBaseTdFiles", + ], +) + +filegroup( + name = "IRDLPyFiles", + srcs = [ + "mlir/dialects/irdl.py", + ":IRDLOpsPyGen", + ], +) + ##---------------------------------------------------------------------------## # Math dialect. ##---------------------------------------------------------------------------##