[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