From 1c47c3525070112125bdd3d81f22aa67e46a5233 Mon Sep 17 00:00:00 2001
From: Yi Zhang <cathyzhyi@google.com>
Date: Mon, 14 Jul 2025 11:47:48 -0400
Subject: [PATCH] [bazel] Add targets for transform.tune python extension

---
 .../mlir/python/BUILD.bazel                   | 26 +++++++++++++++++++
 1 file changed, 26 insertions(+)

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.
 ##---------------------------------------------------------------------------##