mlir::python::PyThreadContextEntry Class Reference

Tracks an entry in the thread context stack. More...

#include "Bindings/Python/IRModule.h"

Public Types
enum class	FrameKind { Context , InsertionPoint , Location }

Public Member Functions
	PyThreadContextEntry (FrameKind frameKind, nanobind::object context, nanobind::object insertionPoint, nanobind::object location)
PyMlirContext *	getContext ()
PyInsertionPoint *	getInsertionPoint ()
PyLocation *	getLocation ()
FrameKind	getFrameKind ()

Static Public Member Functions
static PyMlirContext *	getDefaultContext ()
	Gets the top of stack context and return nullptr if not defined. More...
static PyInsertionPoint *	getDefaultInsertionPoint ()
	Gets the top of stack insertion point and return nullptr if not defined. More...
static PyLocation *	getDefaultLocation ()
	Gets the top of stack location and returns nullptr if not defined. More...
static PyThreadContextEntry *	getTopOfStack ()
	Stack management. More...
static nanobind::object	pushContext (nanobind::object context)
static void	popContext (PyMlirContext &context)
static nanobind::object	pushInsertionPoint (nanobind::object insertionPoint)
static void	popInsertionPoint (PyInsertionPoint &insertionPoint)
static nanobind::object	pushLocation (nanobind::object location)
static void	popLocation (PyLocation &location)
static std::vector< PyThreadContextEntry > &	getStack ()
	Gets the thread local stack. More...

Detailed Description

Tracks an entry in the thread context stack.

New entries are pushed onto here for each with block that activates a new InsertionPoint, Context or Location.

Pushing either a Location or InsertionPoint also pushes its associated Context. Pushing a Context will not modify the Location or InsertionPoint unless if they are from a different context, in which case, they are cleared.

Definition at line 107 of file IRModule.h.

Member Enumeration Documentation

FrameKind

enum mlir::python::PyThreadContextEntry::FrameKind

strong

Enumerator
Context
InsertionPoint
Location

Definition at line 109 of file IRModule.h.

Constructor & Destructor Documentation

PyThreadContextEntry()

mlir::python::PyThreadContextEntry::PyThreadContextEntry ( FrameKind  frameKind, nanobind::object  context, nanobind::object  insertionPoint, nanobind::object  location  )

inline

Definition at line 115 of file IRModule.h.

Member Function Documentation

getContext()

PyMlirContext * PyThreadContextEntry::getContext

(

)

Definition at line 843 of file IRCore.cpp.

getDefaultContext()

PyMlirContext * PyThreadContextEntry::getDefaultContext ( )

static

Gets the top of stack context and return nullptr if not defined.

Definition at line 861 of file IRCore.cpp.

Referenced by mlir::python::DefaultingPyMlirContext::resolve().

getDefaultInsertionPoint()

PyInsertionPoint * PyThreadContextEntry::getDefaultInsertionPoint ( )

static

Gets the top of stack insertion point and return nullptr if not defined.

Definition at line 866 of file IRCore.cpp.

Referenced by maybeInsertOperation().

getDefaultLocation()

PyLocation * PyThreadContextEntry::getDefaultLocation ( )

static

Gets the top of stack location and returns nullptr if not defined.

Definition at line 871 of file IRCore.cpp.

Referenced by mlir::python::DefaultingPyLocation::resolve().

getFrameKind()

FrameKind mlir::python::PyThreadContextEntry::getFrameKind ( )

inline

Definition at line 133 of file IRModule.h.

getInsertionPoint()

PyInsertionPoint * PyThreadContextEntry::getInsertionPoint

(

)

Definition at line 849 of file IRCore.cpp.

getLocation()

PyLocation * PyThreadContextEntry::getLocation

(

)

Definition at line 855 of file IRCore.cpp.

getStack()

std::vector< PyThreadContextEntry > & PyThreadContextEntry::getStack ( )

static

Gets the thread local stack.

Definition at line 809 of file IRCore.cpp.

getTopOfStack()

PyThreadContextEntry * PyThreadContextEntry::getTopOfStack ( )

static

Stack management.

Definition at line 814 of file IRCore.cpp.

popContext()

void PyThreadContextEntry::popContext ( PyMlirContext &  context )

static

Definition at line 883 of file IRCore.cpp.

Referenced by mlir::python::PyMlirContext::contextExit().

popInsertionPoint()

void PyThreadContextEntry::popInsertionPoint ( PyInsertionPoint &  insertionPoint )

static

Definition at line 906 of file IRCore.cpp.

Referenced by mlir::python::PyInsertionPoint::contextExit().

popLocation()

void PyThreadContextEntry::popLocation ( PyLocation &  location )

static

Definition at line 926 of file IRCore.cpp.

Referenced by mlir::python::PyLocation::contextExit().

pushContext()

nb::object PyThreadContextEntry::pushContext ( nanobind::object  context )

static

Definition at line 876 of file IRCore.cpp.

Referenced by mlir::python::PyMlirContext::contextEnter().

pushInsertionPoint()

nb::object PyThreadContextEntry::pushInsertionPoint ( nanobind::object  insertionPoint )

static

Definition at line 894 of file IRCore.cpp.

References mlir::python::PyInsertionPoint::getBlock(), mlir::python::BaseContextObject::getContext(), mlir::python::PyObjectRef< T >::getObject(), and mlir::python::PyBlock::getParentOperation().

Referenced by mlir::python::PyInsertionPoint::contextEnter().

pushLocation()

nb::object PyThreadContextEntry::pushLocation ( nanobind::object  location )

static

Definition at line 917 of file IRCore.cpp.

References mlir::python::BaseContextObject::getContext(), and mlir::python::PyObjectRef< T >::getObject().

Referenced by mlir::python::PyLocation::contextEnter().

---

The documentation for this class was generated from the following files:

• lib/Bindings/Python/IRModule.h
• lib/Bindings/Python/IRCore.cpp