[llvm] [bazel] Add targets for transform.tune python extension (PR #148656)

Yi Zhang via llvm-commits llvm-commits at lists.llvm.org
Mon Jul 14 08:53:48 PDT 2025


https://github.com/cathyzhyi created https://github.com/llvm/llvm-project/pull/148656

None

>From 1c47c3525070112125bdd3d81f22aa67e46a5233 Mon Sep 17 00:00:00 2001
From: Yi Zhang <cathyzhyi at 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.
 ##---------------------------------------------------------------------------##



More information about the llvm-commits mailing list