From 93a6b49d38af236a964492ac6de71d71e93a89a6 Mon Sep 17 00:00:00 2001 From: Alex Zinenko Date: Thu, 30 Sep 2021 15:09:30 +0200 Subject: [PATCH] [mlir][python] provide bindings for ops from the sparse_tensor dialect Previously, the dialect was exposed for linking and pass management purposes, but we did not generate op classes for it. Generate them. Reviewed By: nicolasvasilache Differential Revision: https://reviews.llvm.org/D110819 --- mlir/python/CMakeLists.txt | 1 + mlir/python/mlir/dialects/SparseTensorOps.td | 15 +++++++++++++++ mlir/python/mlir/dialects/sparse_tensor.py | 1 + 3 files changed, 17 insertions(+) create mode 100644 mlir/python/mlir/dialects/SparseTensorOps.td diff --git a/mlir/python/CMakeLists.txt b/mlir/python/CMakeLists.txt index 2ab3a9a..eb7e1e4 100644 --- a/mlir/python/CMakeLists.txt +++ b/mlir/python/CMakeLists.txt @@ -128,6 +128,7 @@ declare_mlir_dialect_python_bindings( declare_mlir_dialect_python_bindings( ADD_TO_PARENT MLIRPythonSources.Dialects ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" + TD_FILE dialects/SparseTensorOps.td SOURCES dialects/sparse_tensor.py DIALECT_NAME sparse_tensor) diff --git a/mlir/python/mlir/dialects/SparseTensorOps.td b/mlir/python/mlir/dialects/SparseTensorOps.td new file mode 100644 index 0000000..b3b4846 --- /dev/null +++ b/mlir/python/mlir/dialects/SparseTensorOps.td @@ -0,0 +1,15 @@ +//===-- SparseTensorOps.td - Entry point for bindings ------*- 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 +// +//===----------------------------------------------------------------------===// + +#ifndef PYTHON_BINDINGS_SPARSE_TENSOR_OPS +#define PYTHON_BINDINGS_SPARSE_TENSOR_OPS + +include "mlir/Bindings/Python/Attributes.td" +include "mlir/Dialect/SparseTensor/IR/SparseTensorOps.td" + +#endif diff --git a/mlir/python/mlir/dialects/sparse_tensor.py b/mlir/python/mlir/dialects/sparse_tensor.py index 4a89ef8..4f6b675 100644 --- a/mlir/python/mlir/dialects/sparse_tensor.py +++ b/mlir/python/mlir/dialects/sparse_tensor.py @@ -2,5 +2,6 @@ # See https://llvm.org/LICENSE.txt for license information. # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +from ._sparse_tensor_ops_gen import * from .._mlir_libs._mlir.dialects.sparse_tensor import * from .._mlir_libs import _mlirSparseTensorPasses as _cextSparseTensorPasses -- 2.7.4