diff --git a/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel index 1d00afb4e8a2b..39510dd3415a9 100644 --- a/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel +++ b/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel @@ -700,6 +700,32 @@ filegroup( ], ) +##---------------------------------------------------------------------------## +# Tune dialect. +##---------------------------------------------------------------------------## + +gentbl_filegroup( + name = "TuneTransformOpsPyGen", + tbl_outs = {"mlir/dialects/_transform_tune_extension_ops_gen.py": [ + "-gen-python-op-bindings", + "-bind-dialect=transform", + "-dialect-extension=transform_tune_extension", + ]}, + tblgen = "//third_party/llvm/llvm-project/mlir:mlir-tblgen", + td_file = "mlir/dialects/TransformTuneExtensionOps.td", + deps = [ + "//third_party/llvm/llvm-project/mlir:TransformTuneExtensionTdFiles", + ], +) + +filegroup( + name = "TunePyFiles", + srcs = [ + "mlir/dialects/transform/tune.py", + ":TuneTransformOpsPyGen", + ], +) + ##---------------------------------------------------------------------------## # PythonTest dialect. ##---------------------------------------------------------------------------##