diff --git a/mlir/python/CMakeLists.txt b/mlir/python/CMakeLists.txt --- a/mlir/python/CMakeLists.txt +++ b/mlir/python/CMakeLists.txt @@ -134,6 +134,15 @@ _mlir_libs/_mlir/dialects/transform/__init__.pyi DIALECT_NAME transform) +declare_mlir_dialect_extension_python_bindings( + ADD_TO_PARENT MLIRPythonSources.Dialects + ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" + TD_FILE dialects/BufferizationTransformOps.td + SOURCES + dialects/transform/bufferization.py + DIALECT_NAME transform + EXTENSION_NAME bufferization_transform) + declare_mlir_dialect_extension_python_bindings( ADD_TO_PARENT MLIRPythonSources.Dialects ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" diff --git a/mlir/python/mlir/dialects/BufferizationTransformOps.td b/mlir/python/mlir/dialects/BufferizationTransformOps.td new file mode 100644 --- /dev/null +++ b/mlir/python/mlir/dialects/BufferizationTransformOps.td @@ -0,0 +1,21 @@ +//===-- BufferizationTransformOps.td -----------------------*- tablegen -*-===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// +// +// Entry point of the Python bindings generator for the transform ops provided +// by the bufferization dialect. +// +//===----------------------------------------------------------------------===// + + +#ifndef PYTHON_BINDINGS_BUFFERIZATION_TRANSFORM_OPS +#define PYTHON_BINDINGS_BUFFERIZATION_TRANSFORM_OPS + +include "mlir/Bindings/Python/Attributes.td" +include "mlir/Dialect/Bufferization/TransformOps/BufferizationTransformOps.td" + +#endif // PYTHON_BINDINGS_BUFFERIZATION_TRANSFORM_OPS diff --git a/mlir/python/mlir/dialects/transform/bufferization.py b/mlir/python/mlir/dialects/transform/bufferization.py new file mode 100644 --- /dev/null +++ b/mlir/python/mlir/dialects/transform/bufferization.py @@ -0,0 +1,5 @@ +# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +# See https://llvm.org/LICENSE.txt for license information. +# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception + +from .._bufferization_transform_ops_gen import *