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)17PYBIND11_MODULE(_mlirSparseTensorPasses, m) { 18 m.doc() = "MLIR SparseTensor Dialect Passes"; 19 20 // Register all SparseTensor passes on load. 21 mlirRegisterSparseTensorPasses(); 22 } 23