1 //===- SparseTensorPasses.cpp - Pybind module for the SparseTensor passes -===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include "mlir-c/Dialect/SparseTensor.h"
10 
11 #include <pybind11/pybind11.h>
12 
13 // -----------------------------------------------------------------------------
14 // Module initialization.
15 // -----------------------------------------------------------------------------
16 
PYBIND11_MODULE(_mlirSparseTensorPasses,m)17 PYBIND11_MODULE(_mlirSparseTensorPasses, m) {
18   m.doc() = "MLIR SparseTensor Dialect Passes";
19 
20   // Register all SparseTensor passes on load.
21   mlirRegisterSparseTensorPasses();
22 }
23