mirror of
https://github.com/intel/llvm.git
synced 2026-01-13 11:02:04 +08:00
efd96afedf
This a reland of https://github.com/llvm/llvm-project/pull/155741 which was reverted at https://github.com/llvm/llvm-project/pull/157831. This version is narrower in scope - it only turns on automatic stub generation for `MLIRPythonExtension.Core._mlir` and **does not do anything automatically**. Specifically, the only CMake code added to `AddMLIRPython.cmake` is the `mlir_generate_type_stubs` function which is then used only in a manual way. The API for `mlir_generate_type_stubs` is: ``` Arguments: MODULE_NAME: The fully-qualified name of the extension module (used for importing in python). DEPENDS_TARGETS: List of targets these type stubs depend on being built; usually corresponding to the specific extension module (e.g., something like StandalonePythonModules.extension._standaloneDialectsNanobind.dso) and the core bindings extension module (e.g., something like StandalonePythonModules.extension._mlir.dso). OUTPUT_DIR: The root output directory to emit the type stubs into. OUTPUTS: List of expected outputs. DEPENDS_TARGET_SRC_DEPS: List of cpp sources for extension library (for generating a DEPFILE). IMPORT_PATHS: List of paths to add to PYTHONPATH for stubgen. PATTERN_FILE: (Optional) Pattern file (see https://nanobind.readthedocs.io/en/latest/typing.html#pattern-files). Outputs: NB_STUBGEN_CUSTOM_TARGET: The target corresponding to generation which other targets can depend on. ``` Downstream users should use `mlir_generate_type_stubs` in coordination with `declare_mlir_python_sources` to turn on stub generation for their own downstream dialect extensions and upstream dialect extensions if they so choose. Standalone example shows an example. Note, downstream will also need to set `-DMLIR_PYTHON_PACKAGE_PREFIX=...` correctly for their bindings.
106 lines
3.9 KiB
C++
106 lines
3.9 KiB
C++
//===- DialectPDL.cpp - 'pdl' dialect submodule ---------------------------===//
|
|
//
|
|
// 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 "mlir-c/Dialect/PDL.h"
|
|
#include "mlir-c/IR.h"
|
|
#include "mlir/Bindings/Python/Nanobind.h"
|
|
#include "mlir/Bindings/Python/NanobindAdaptors.h"
|
|
|
|
namespace nb = nanobind;
|
|
using namespace llvm;
|
|
using namespace mlir;
|
|
using namespace mlir::python;
|
|
using namespace mlir::python::nanobind_adaptors;
|
|
|
|
static void populateDialectPDLSubmodule(const nanobind::module_ &m) {
|
|
//===-------------------------------------------------------------------===//
|
|
// PDLType
|
|
//===-------------------------------------------------------------------===//
|
|
|
|
auto pdlType = mlir_type_subclass(m, "PDLType", mlirTypeIsAPDLType);
|
|
|
|
//===-------------------------------------------------------------------===//
|
|
// AttributeType
|
|
//===-------------------------------------------------------------------===//
|
|
|
|
auto attributeType =
|
|
mlir_type_subclass(m, "AttributeType", mlirTypeIsAPDLAttributeType);
|
|
attributeType.def_classmethod(
|
|
"get",
|
|
[](const nb::object &cls, MlirContext ctx) {
|
|
return cls(mlirPDLAttributeTypeGet(ctx));
|
|
},
|
|
"Get an instance of AttributeType in given context.", nb::arg("cls"),
|
|
nb::arg("context") = nb::none());
|
|
|
|
//===-------------------------------------------------------------------===//
|
|
// OperationType
|
|
//===-------------------------------------------------------------------===//
|
|
|
|
auto operationType =
|
|
mlir_type_subclass(m, "OperationType", mlirTypeIsAPDLOperationType);
|
|
operationType.def_classmethod(
|
|
"get",
|
|
[](const nb::object &cls, MlirContext ctx) {
|
|
return cls(mlirPDLOperationTypeGet(ctx));
|
|
},
|
|
"Get an instance of OperationType in given context.", nb::arg("cls"),
|
|
nb::arg("context") = nb::none());
|
|
|
|
//===-------------------------------------------------------------------===//
|
|
// RangeType
|
|
//===-------------------------------------------------------------------===//
|
|
|
|
auto rangeType = mlir_type_subclass(m, "RangeType", mlirTypeIsAPDLRangeType);
|
|
rangeType.def_classmethod(
|
|
"get",
|
|
[](const nb::object &cls, MlirType elementType) {
|
|
return cls(mlirPDLRangeTypeGet(elementType));
|
|
},
|
|
"Gets an instance of RangeType in the same context as the provided "
|
|
"element type.",
|
|
nb::arg("cls"), nb::arg("element_type"));
|
|
rangeType.def_property_readonly(
|
|
"element_type",
|
|
[](MlirType type) { return mlirPDLRangeTypeGetElementType(type); },
|
|
nb::sig(
|
|
"def element_type(self) -> " MAKE_MLIR_PYTHON_QUALNAME("ir.Type")),
|
|
"Get the element type.");
|
|
|
|
//===-------------------------------------------------------------------===//
|
|
// TypeType
|
|
//===-------------------------------------------------------------------===//
|
|
|
|
auto typeType = mlir_type_subclass(m, "TypeType", mlirTypeIsAPDLTypeType);
|
|
typeType.def_classmethod(
|
|
"get",
|
|
[](const nb::object &cls, MlirContext ctx) {
|
|
return cls(mlirPDLTypeTypeGet(ctx));
|
|
},
|
|
"Get an instance of TypeType in given context.", nb::arg("cls"),
|
|
nb::arg("context") = nb::none());
|
|
|
|
//===-------------------------------------------------------------------===//
|
|
// ValueType
|
|
//===-------------------------------------------------------------------===//
|
|
|
|
auto valueType = mlir_type_subclass(m, "ValueType", mlirTypeIsAPDLValueType);
|
|
valueType.def_classmethod(
|
|
"get",
|
|
[](const nb::object &cls, MlirContext ctx) {
|
|
return cls(mlirPDLValueTypeGet(ctx));
|
|
},
|
|
"Get an instance of TypeType in given context.", nb::arg("cls"),
|
|
nb::arg("context") = nb::none());
|
|
}
|
|
|
|
NB_MODULE(_mlirDialectsPDL, m) {
|
|
m.doc() = "MLIR PDL dialect.";
|
|
populateDialectPDLSubmodule(m);
|
|
}
|