From b96bd025b35761ae181da7e1796708c46e59f5d5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ingo=20M=C3=BCller?= <ingomueller@google.com>
Date: Tue, 18 Jul 2023 15:02:32 +0000
Subject: [PATCH] [mlir][transform][gpu][python] Add .td file for bindings.

Reviewed By: ftynse

Differential Revision: https://reviews.llvm.org/D155602
---
 mlir/python/CMakeLists.txt                   |  9 +++++++++
 mlir/python/mlir/dialects/GPUTransformOps.td | 20 ++++++++++++++++++++
 mlir/python/mlir/dialects/transform/gpu.py   |  5 +++++
 3 files changed, 34 insertions(+)
 create mode 100644 mlir/python/mlir/dialects/GPUTransformOps.td
 create mode 100644 mlir/python/mlir/dialects/transform/gpu.py

diff --git a/mlir/python/CMakeLists.txt b/mlir/python/CMakeLists.txt
index 22a55dbb66aa..e5d37b228986 100644
--- a/mlir/python/CMakeLists.txt
+++ b/mlir/python/CMakeLists.txt
@@ -143,6 +143,15 @@ declare_mlir_dialect_extension_python_bindings(
   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
index 000000000000..08bd9537b459
--- /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
index 000000000000..8c3de0de7ea3
--- /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 *