Files
llvm/mlir/lib/Bindings/Python/Dialects.h
Alex Zinenko 66d4090d9b [mlir] Introduce Python bindings for the quantization dialect
So far, only the custom dialect types are exposed.

The build and packaging is same as for Linalg and SparseTensor, and in
need of refactoring that is beyond the scope of this patch.

Reviewed By: stellaraccident

Differential Revision: https://reviews.llvm.org/D116605
2022-01-05 16:26:31 +01:00

27 lines
941 B
C++

//===- Dialects.h - Declaration for dialect submodule factories -----------===//
//
// 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
//
//===----------------------------------------------------------------------===//
#ifndef MLIR_BINDINGS_PYTHON_DIALECTS_H
#define MLIR_BINDINGS_PYTHON_DIALECTS_H
#include <pybind11/pybind11.h>
namespace mlir {
namespace python {
void populateDialectLinalgSubmodule(pybind11::module m);
void populateDialectSparseTensorSubmodule(const pybind11::module &m,
const pybind11::module &irModule);
void populateDialectQuantSubmodule(const pybind11::module &m,
const pybind11::module &irModule);
} // namespace python
} // namespace mlir
#endif // MLIR_BINDINGS_PYTHON_DIALECTS_H