58d12332a4
First set of "boilerplate" to get sparse tensor passes available through CAPI and Python. Reviewed By: stellaraccident Differential Revision: https://reviews.llvm.org/D102362
23 lines
813 B
C++
23 lines
813 B
C++
//===- SparseTensorPasses.cpp - Pybind module for the SparseTensor passes -===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir-c/Dialect/SparseTensor.h"
|
|
|
|
#include <pybind11/pybind11.h>
|
|
|
|
// -----------------------------------------------------------------------------
|
|
// Module initialization.
|
|
// -----------------------------------------------------------------------------
|
|
|
|
PYBIND11_MODULE(_mlirSparseTensorPasses, m) {
|
|
m.doc() = "MLIR SparseTensor Dialect Passes";
|
|
|
|
// Register all SparseTensor passes on load.
|
|
mlirRegisterSparseTensorPasses();
|
|
}
|