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

This commit is contained in:
Yi Zhang
2025-07-14 12:19:32 -04:00
committed by GitHub
parent d5ac1b5e28
commit 96096550a1

View File

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