Lines Matching defs:PyMlirContext

580         PyMlirContext::forContext(mlirOperationGetContext(owner));
624 // PyMlirContext
627 PyMlirContext::PyMlirContext(MlirContext context) : context(context) {
634 PyMlirContext::~PyMlirContext() {
646 nb::object PyMlirContext::getCapsule() {
650 nb::object PyMlirContext::createFromCapsule(nb::object capsule) {
657 PyMlirContextRef PyMlirContext::forContext(MlirContext context) {
664     PyMlirContext *unownedContextWrapper = new PyMlirContext(context);
675 nb::ft_mutex PyMlirContext::live_contexts_mutex;
677 PyMlirContext::LiveContextMap &PyMlirContext::getLiveContexts() {
682 size_t PyMlirContext::getLiveCount() {
687 size_t PyMlirContext::getLiveOperationCount() {
692 std::vector<PyOperation *> PyMlirContext::getLiveOperationObjects() {
700 size_t PyMlirContext::clearLiveOperations() {
713 void PyMlirContext::clearOperation(MlirOperation op) {
727 void PyMlirContext::clearOperationsInside(PyOperationBase &op) {
747 void PyMlirContext::clearOperationsInside(MlirOperation op) {
752 void PyMlirContext::clearOperationAndInside(PyOperationBase &op) {
763 size_t PyMlirContext::getLiveModuleCount() { return liveModules.size(); }
765 nb::object PyMlirContext::contextEnter(nb::object context) {
769 void PyMlirContext::contextExit(const nb::object &excType,
775 nb::object PyMlirContext::attachDiagnosticHandler(nb::object callback) {
825 MlirLogicalResult PyMlirContext::ErrorCapture::handler(MlirDiagnostic diag,
839 PyMlirContext &DefaultingPyMlirContext::resolve() {
840   PyMlirContext *context = PyThreadContextEntry::getDefaultContext();
888 PyMlirContext *PyThreadContextEntry::getContext() {
891   return nb::cast<PyMlirContext *>(context);
906 PyMlirContext *PyThreadContextEntry::getDefaultContext() {
928 void PyThreadContextEntry::popContext(PyMlirContext &context) {
1028   return PyLocation(PyMlirContext::forContext(context), loc);
1104   return PyLocation(PyMlirContext::forContext(mlirLocationGetContext(rawLoc)),
1147   PyMlirContextRef contextRef = PyMlirContext::forContext(context);
1273   PyMlirContext::ErrorCapture errors(contextRef);
1428   PyMlirContext::ErrorCapture errors(op.getContext());
1463   return forOperation(PyMlirContext::forContext(rawCtxt), rawOperation)
2178       PyMlirContext::forContext(mlirAttributeGetContext(rawAttr)), rawAttr);
2209   return PyType(PyMlirContext::forContext(mlirTypeGetContext(rawType)),
2267       PyOperation::forOperation(PyMlirContext::forContext(ctx), owner);
2736   nb::class_<PyMlirContext>(m, "_BaseContext")
2738            [](PyMlirContext &self) {
2740              new (&self) PyMlirContext(context);
2742       .def_static("_get_live_count", &PyMlirContext::getLiveCount)
2744            [](PyMlirContext &self) {
2745              PyMlirContextRef ref = PyMlirContext::forContext(self.get());
2748       .def("_get_live_operation_count", &PyMlirContext::getLiveOperationCount)
2750            &PyMlirContext::getLiveOperationObjects)
2751       .def("_clear_live_operations", &PyMlirContext::clearLiveOperations)
2754                &PyMlirContext::clearOperationsInside))
2755       .def("_get_live_module_count", &PyMlirContext::getLiveModuleCount)
2756       .def_prop_ro(MLIR_PYTHON_CAPI_PTR_ATTR, &PyMlirContext::getCapsule)
2757       .def(MLIR_PYTHON_CAPI_FACTORY_ATTR, &PyMlirContext::createFromCapsule)
2758       .def("__enter__", &PyMlirContext::contextEnter)
2759       .def("__exit__", &PyMlirContext::contextExit, nb::arg("exc_type").none(),
2772           [](PyMlirContext &self) { return PyDialects(self.getRef()); },
2775           "d", [](PyMlirContext &self) { return PyDialects(self.getRef()); },
2779           [=](PyMlirContext &self, std::string &name) {
2792           [](PyMlirContext &self) -> bool {
2795           [](PyMlirContext &self, bool value) {
2798       .def("attach_diagnostic_handler", &PyMlirContext::attachDiagnosticHandler,
2803           [](PyMlirContext &self, bool enable) {
2809           [](PyMlirContext &self, std::string &name) {
2816           [](PyMlirContext &self, PyDialectRegistry &registry) {
2821                    &PyMlirContext::setEmitErrorDiagnostics,
2825       .def("load_all_available_dialects", [](PyMlirContext &self) {
3029             PyMlirContext::ErrorCapture errors(context->getRef());
3041             PyMlirContext::ErrorCapture errors(context->getRef());
3621             PyMlirContext::ErrorCapture errors(context->getRef());
3739             PyMlirContext::ErrorCapture errors(context->getRef());