[mlir][transform][gpu][python] Add .td file for bindings.

Reviewed By: ftynse

Differential Revision: https://reviews.llvm.org/D155602
This commit is contained in:
Ingo Müller
2023-07-18 15:02:32 +00:00
parent 4f60815114
commit b96bd025b3
3 changed files with 34 additions and 0 deletions

View File

@@ -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"

View File

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

View File

@@ -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 *