diff --git a/mlir/python/CMakeLists.txt b/mlir/python/CMakeLists.txt --- a/mlir/python/CMakeLists.txt +++ b/mlir/python/CMakeLists.txt @@ -143,6 +143,15 @@ 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" + TD_FILE dialects/GPUTransformOps.td + SOURCES + dialects/transform/gpu.py + DIALECT_NAME transform + EXTENSION_NAME gpu_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/GPUTransformOps.td b/mlir/python/mlir/dialects/GPUTransformOps.td new file mode 100644 --- /dev/null +++ b/mlir/python/mlir/dialects/GPUTransformOps.td @@ -0,0 +1,20 @@ +//===-- GPUTransformOps.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 GPU dialect. +// +//===----------------------------------------------------------------------===// + + +#ifndef PYTHON_BINDINGS_GPU_TRANSFORM_OPS +#define PYTHON_BINDINGS_GPU_TRANSFORM_OPS + +include "mlir/Dialect/GPU/TransformOps/GPUTransformOps.td" + +#endif // PYTHON_BINDINGS_GPU_TRANSFORM_OPS diff --git a/mlir/python/mlir/dialects/transform/gpu.py b/mlir/python/mlir/dialects/transform/gpu.py new file mode 100644 --- /dev/null +++ b/mlir/python/mlir/dialects/transform/gpu.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 .._gpu_transform_ops_gen import *