840cc16a8d
* `_OperationBase.walk` was missing a default. * `MLIRError` is now fully defined in C++. The monkey-patching previously done in `_site_initialize` was opaque to type checkers.
58 lines
2.0 KiB
C++
58 lines
2.0 KiB
C++
//===- MainModule.cpp - Main pybind module --------------------------------===//
|
|
//
|
|
// 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 "Pass.h"
|
|
#include "Rewrite.h"
|
|
#include "mlir/Bindings/Python/Globals.h"
|
|
#include "mlir/Bindings/Python/IRAttributes.h"
|
|
#include "mlir/Bindings/Python/IRCore.h"
|
|
#include "mlir/Bindings/Python/IRTypes.h"
|
|
#include "mlir/Bindings/Python/Nanobind.h"
|
|
|
|
namespace nb = nanobind;
|
|
using namespace mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN;
|
|
|
|
namespace mlir {
|
|
namespace python {
|
|
namespace MLIR_BINDINGS_PYTHON_DOMAIN {
|
|
void populateIRAffine(nb::module_ &m);
|
|
void populateIRAttributes(nb::module_ &m);
|
|
void populateIRInterfaces(nb::module_ &m);
|
|
void populateIRTypes(nb::module_ &m);
|
|
void populateIRCore(nb::module_ &m);
|
|
void populateRoot(nb::module_ &m);
|
|
} // namespace MLIR_BINDINGS_PYTHON_DOMAIN
|
|
} // namespace python
|
|
} // namespace mlir
|
|
|
|
// -----------------------------------------------------------------------------
|
|
// Module initialization.
|
|
// -----------------------------------------------------------------------------
|
|
NB_MODULE(_mlir, m) {
|
|
// disable leak warnings which tend to be false positives.
|
|
nb::set_leak_warnings(false);
|
|
|
|
m.doc() = "MLIR Python Native Extension";
|
|
populateRoot(m);
|
|
// Define and populate IR submodule.
|
|
auto irModule = m.def_submodule("ir", "MLIR IR Bindings");
|
|
populateIRCore(irModule);
|
|
populateIRAffine(irModule);
|
|
populateIRAttributes(irModule);
|
|
populateIRInterfaces(irModule);
|
|
populateIRTypes(irModule);
|
|
|
|
auto rewriteModule = m.def_submodule("rewrite", "MLIR Rewrite Bindings");
|
|
populateRewriteSubmodule(rewriteModule);
|
|
|
|
// Define and populate PassManager submodule.
|
|
auto passManagerModule =
|
|
m.def_submodule("passmanager", "MLIR Pass Management Bindings");
|
|
populatePassManagerSubmodule(passManagerModule);
|
|
}
|