23 lines · cpp
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-exception6//7//===----------------------------------------------------------------------===//8 9#include "mlir-c/Dialect/SparseTensor.h"10 11#include "mlir/Bindings/Python/Nanobind.h"12 13// -----------------------------------------------------------------------------14// Module initialization.15// -----------------------------------------------------------------------------16 17NB_MODULE(_mlirSparseTensorPasses, m) {18 m.doc() = "MLIR SparseTensor Dialect Passes";19 20 // Register all SparseTensor passes on load.21 mlirRegisterSparseTensorPasses();22}23