This is an archive of the discontinued LLVM Phabricator instance.

Differential D110819

[mlir][python] provide bindings for ops from the sparse_tensor dialect
ClosedPublic

Authored by ftynse on Sep 30 2021, 6:10 AM.

Details

Reviewers
nicolasvasilache
aartbik
Commits
rG93a6b49d38af: [mlir][python] provide bindings for ops from the sparse_tensor dialect
Summary

Previously, the dialect was exposed for linking and pass management purposes,
but we did not generate op classes for it. Generate them.

Diff Detail

Event Timeline

ftynse created this revision.Sep 30 2021, 6:10 AM
Herald added subscribers: wenzhicui, wrengr, Chia-hungDuan and 19 others. · View Herald TranscriptSep 30 2021, 6:10 AM
ftynse requested review of this revision.Sep 30 2021, 6:10 AM
Herald added a project: Restricted Project. · View Herald TranscriptSep 30 2021, 6:10 AM
Herald added a subscriber: stephenneuendorffer. · View Herald Transcript
Harbormaster completed remote builds in B126539: Diff 376173.Sep 30 2021, 6:21 AM
nicolasvasilache accepted this revision.Sep 30 2021, 6:37 AM

!!

This revision is now accepted and ready to land.Sep 30 2021, 6:37 AM
Closed by commit rG93a6b49d38af: [mlir][python] provide bindings for ops from the sparse_tensor dialect (authored by ftynse). · Explain WhySep 30 2021, 6:53 AM
This revision was automatically updated to reflect the committed changes.
ftynse added a commit: rG93a6b49d38af: [mlir][python] provide bindings for ops from the sparse_tensor dialect.

Revision Contents

PathSize
mlir/
python/
1 line
mlir/
dialects/
15 lines
1 line

Diff 376173

mlir/python/CMakeLists.txt

Show First 20 Lines • Show All 122 Lines • ▼ Show 20 Linesdeclare_mlir_dialect_python_bindings(
ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir"
TD_FILE dialects/ShapeOps.td TD_FILE dialects/ShapeOps.td
SOURCES dialects/shape.py SOURCES dialects/shape.py
DIALECT_NAME shape) DIALECT_NAME shape)
declare_mlir_dialect_python_bindings( declare_mlir_dialect_python_bindings(
ADD_TO_PARENT MLIRPythonSources.Dialects ADD_TO_PARENT MLIRPythonSources.Dialects
ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir"
TD_FILE dialects/SparseTensorOps.td
SOURCES dialects/sparse_tensor.py SOURCES dialects/sparse_tensor.py
DIALECT_NAME sparse_tensor) DIALECT_NAME sparse_tensor)
declare_mlir_dialect_python_bindings( declare_mlir_dialect_python_bindings(
ADD_TO_PARENT MLIRPythonSources.Dialects ADD_TO_PARENT MLIRPythonSources.Dialects
ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir"
TD_FILE dialects/StandardOps.td TD_FILE dialects/StandardOps.td
SOURCES dialects/std.py SOURCES dialects/std.py
▲ Show 20 Lines • Show All 193 Lines • Show Last 20 Lines

mlir/python/mlir/dialects/SparseTensorOps.td

  • This file was added.
//===-- 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

mlir/python/mlir/dialects/sparse_tensor.py

# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. # Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information. # See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception # 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._mlir.dialects.sparse_tensor import *
from .._mlir_libs import _mlirSparseTensorPasses as _cextSparseTensorPasses from .._mlir_libs import _mlirSparseTensorPasses as _cextSparseTensorPasses