MLIR
19.0.0git
|
#include "Bindings/Python/IRModule.h"
Classes | |
struct | ErrorCapture |
RAII object that captures any error diagnostics emitted to the provided context. More... | |
Public Member Functions | |
PyMlirContext ()=delete | |
PyMlirContext (const PyMlirContext &)=delete | |
PyMlirContext (PyMlirContext &&)=delete | |
~PyMlirContext () | |
MlirContext | get () |
Accesses the underlying MlirContext. More... | |
PyMlirContextRef | getRef () |
Gets a strong reference to this context, which will ensure it is kept alive for the life of the reference. More... | |
pybind11::object | getCapsule () |
Gets a capsule wrapping the void* within the MlirContext. More... | |
std::vector< PyOperation * > | getLiveOperationObjects () |
Get a list of Python objects which are still in the live context map. More... | |
size_t | getLiveOperationCount () |
Gets the count of live operations associated with this context. More... | |
size_t | clearLiveOperations () |
Clears the live operations map, returning the number of entries which were invalidated. More... | |
void | clearOperation (MlirOperation op) |
Removes an operation from the live operations map and sets it invalid. More... | |
void | clearOperationsInside (PyOperationBase &op) |
Clears all operations nested inside the given op using clearOperation(MlirOperation) . More... | |
void | clearOperationsInside (MlirOperation op) |
size_t | getLiveModuleCount () |
Gets the count of live modules associated with this context. More... | |
pybind11::object | contextEnter () |
Enter and exit the context manager. More... | |
void | contextExit (const pybind11::object &excType, const pybind11::object &excVal, const pybind11::object &excTb) |
pybind11::object | attachDiagnosticHandler (pybind11::object callback) |
Attaches a Python callback as a diagnostic handler, returning a registration object (internally a PyDiagnosticHandler). More... | |
void | setEmitErrorDiagnostics (bool value) |
Controls whether error diagnostics should be propagated to diagnostic handlers, instead of being captured by ErrorCapture . More... | |
Static Public Member Functions | |
static PyMlirContext * | createNewContextForInit () |
For the case of a python init (py::init) method, pybind11 is quite strict about needing to return a pointer that is not yet associated to an py::object. More... | |
static PyMlirContextRef | forContext (MlirContext context) |
Returns a context reference for the singleton PyMlirContext wrapper for the given context. More... | |
static pybind11::object | createFromCapsule (pybind11::object capsule) |
Creates a PyMlirContext from the MlirContext wrapped by a capsule. More... | |
static size_t | getLiveCount () |
Gets the count of live context objects. Used for testing. More... | |
Friends | |
class | PyModule |
class | PyOperation |
Definition at line 162 of file IRModule.h.
|
delete |
Referenced by createNewContextForInit(), and forContext().
|
delete |
|
delete |
PyMlirContext::~PyMlirContext | ( | ) |
Definition at line 588 of file IRCore.cpp.
References mlirContextDestroy().
py::object PyMlirContext::attachDiagnosticHandler | ( | pybind11::object | callback | ) |
Attaches a Python callback as a diagnostic handler, returning a registration object (internally a PyDiagnosticHandler).
Definition at line 699 of file IRCore.cpp.
References get(), mlir::python::PyDiagnostic::invalidate(), mlirContextAttachDiagnosticHandler(), mlirLogicalResultFailure(), and mlirLogicalResultSuccess().
size_t PyMlirContext::clearLiveOperations | ( | ) |
Clears the live operations map, returning the number of entries which were invalidated.
To be used as a safety mechanism so that API end-users can't corrupt by holding references they shouldn't have accessed in the first place.
Definition at line 646 of file IRCore.cpp.
void PyMlirContext::clearOperation | ( | MlirOperation | op | ) |
Removes an operation from the live operations map and sets it invalid.
This is useful for when some non-bindings code destroys the operation and the bindings need to made aware. For example, in the case when pass manager is run.
Definition at line 654 of file IRCore.cpp.
void PyMlirContext::clearOperationsInside | ( | MlirOperation | op | ) |
Definition at line 682 of file IRCore.cpp.
void PyMlirContext::clearOperationsInside | ( | PyOperationBase & | op | ) |
Clears all operations nested inside the given op using clearOperation(MlirOperation)
.
Definition at line 662 of file IRCore.cpp.
pybind11::object PyMlirContext::contextEnter | ( | ) |
Enter and exit the context manager.
Definition at line 689 of file IRCore.cpp.
References mlir::python::PyThreadContextEntry::pushContext().
void PyMlirContext::contextExit | ( | const pybind11::object & | excType, |
const pybind11::object & | excVal, | ||
const pybind11::object & | excTb | ||
) |
Definition at line 693 of file IRCore.cpp.
References mlir::python::PyThreadContextEntry::popContext().
|
static |
Creates a PyMlirContext from the MlirContext wrapped by a capsule.
Note that PyMlirContext instances are uniqued, so the returned object may be a pre-existing object. Ownership of the underlying MlirContext is taken by calling this function.
Definition at line 601 of file IRCore.cpp.
References forContext(), mlirContextIsNull(), mlirPythonCapsuleToContext(), and mlir::python::PyObjectRef< T >::releaseObject().
|
static |
For the case of a python init (py::init) method, pybind11 is quite strict about needing to return a pointer that is not yet associated to an py::object.
Since the forContext() method acts like a pool, possibly returning a recycled context, it does not satisfy this need. The usual way in python to accomplish such a thing is to override new, but that is also not supported by pybind11. Instead, we use this entry point which always constructs a fresh context (which cannot alias an existing one because it is fresh).
Definition at line 608 of file IRCore.cpp.
References mlirContextCreateWithThreading(), and PyMlirContext().
|
static |
Returns a context reference for the singleton PyMlirContext wrapper for the given context.
Definition at line 613 of file IRCore.cpp.
References PyMlirContext().
Referenced by createFromCapsule(), mlir::python::PyLocation::createFromCapsule(), mlir::python::PyOperation::createFromCapsule(), mlir::python::PyType::createFromCapsule(), mlir::python::PyAttribute::createFromCapsule(), mlir::python::PyValue::createFromCapsule(), mlir::python::PyAffineExpr::createFromCapsule(), mlir::python::PyAffineMap::createFromCapsule(), mlir::python::PyIntegerSet::createFromCapsule(), mlir::python::PyModule::forModule(), and mlir::python::PyDiagnostic::getLocation().
|
inline |
Accesses the underlying MlirContext.
Definition at line 184 of file IRModule.h.
Referenced by attachDiagnosticHandler(), mlir::python::PyOpView::buildGeneric(), getCapsule(), mlir::python::PyInferShapedTypeOpInterface::inferReturnTypeComponents(), mlir::python::PyInferTypeOpInterface::inferReturnTypes(), mlir::python::PyConcreteOpInterface< ConcreteIface >::PyConcreteOpInterface(), mlir::python::PySymbolTable::setSymbolName(), mlir::python::PySymbolTable::setVisibility(), and mlir::python::PyMlirContext::ErrorCapture::~ErrorCapture().
py::object PyMlirContext::getCapsule | ( | ) |
Gets a capsule wrapping the void* within the MlirContext.
Definition at line 597 of file IRCore.cpp.
References get(), and mlirPythonContextToCapsule().
|
static |
Gets the count of live context objects. Used for testing.
Definition at line 635 of file IRCore.cpp.
size_t PyMlirContext::getLiveModuleCount | ( | ) |
Gets the count of live modules associated with this context.
Used for testing.
Definition at line 687 of file IRCore.cpp.
size_t PyMlirContext::getLiveOperationCount | ( | ) |
Gets the count of live operations associated with this context.
Used for testing.
Definition at line 637 of file IRCore.cpp.
std::vector< PyOperation * > PyMlirContext::getLiveOperationObjects | ( | ) |
Get a list of Python objects which are still in the live context map.
Definition at line 639 of file IRCore.cpp.
|
inline |
Gets a strong reference to this context, which will ensure it is kept alive for the life of the reference.
Definition at line 188 of file IRModule.h.
|
inline |
Controls whether error diagnostics should be propagated to diagnostic handlers, instead of being captured by ErrorCapture
.
Definition at line 244 of file IRModule.h.
Definition at line 277 of file IRModule.h.
|
friend |
Definition at line 278 of file IRModule.h.