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