mirror of https://github.com/llvm/torch-mlir
36 lines
945 B
C++
36 lines
945 B
C++
|
//===- native.cpp - MLIR Python bindings ----------------------------------===//
|
||
|
|
//
|
||
|
|
// 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 <cstddef>
|
||
|
|
#include <unordered_map>
|
||
|
|
|
||
|
|
#include <pybind11/pybind11.h>
|
||
|
|
#include <pybind11/pytypes.h>
|
||
|
|
#include <pybind11/stl.h>
|
||
|
|
|
||
|
|
namespace py = pybind11;
|
||
|
|
|
||
|
|
namespace npcomp {
|
||
|
|
namespace python {
|
||
|
|
|
||
|
|
// Externs
|
||
|
|
void npcompMlirInitialize();
|
||
|
|
void defineMlirEdscModule(py::module m);
|
||
|
|
|
||
|
|
PYBIND11_MODULE(native, m) {
|
||
|
|
npcompMlirInitialize();
|
||
|
|
m.doc() = "Npcomp native python bindings";
|
||
|
|
|
||
|
|
auto mlir_m = m.def_submodule("mlir", "MLIR interop");
|
||
|
|
auto mlir_edsc_m = mlir_m.def_submodule("edsc");
|
||
|
|
defineMlirEdscModule(mlir_edsc_m);
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace python
|
||
|
|
} // namespace npcomp
|