diff options
Diffstat (limited to 'mlir/lib')
-rw-r--r-- | mlir/lib/Bindings/Python/IRCore.cpp | 6 | ||||
-rw-r--r-- | mlir/lib/CAPI/IR/IR.cpp | 4 |
2 files changed, 10 insertions, 0 deletions
diff --git a/mlir/lib/Bindings/Python/IRCore.cpp b/mlir/lib/Bindings/Python/IRCore.cpp index d961482..002923b 100644 --- a/mlir/lib/Bindings/Python/IRCore.cpp +++ b/mlir/lib/Bindings/Python/IRCore.cpp @@ -2939,6 +2939,12 @@ void mlir::python::populateIRCore(nb::module_ &m) { ss << pool.ptr; return ss.str(); }) + .def_prop_ro( + "is_multithreading_enabled", + [](PyMlirContext &self) { + return mlirContextIsMultithreadingEnabled(self.get()); + }, + "Returns true if multithreading is enabled for this context.") .def( "is_registered_operation", [](PyMlirContext &self, std::string &name) { diff --git a/mlir/lib/CAPI/IR/IR.cpp b/mlir/lib/CAPI/IR/IR.cpp index fbc66bc..1cc555a 100644 --- a/mlir/lib/CAPI/IR/IR.cpp +++ b/mlir/lib/CAPI/IR/IR.cpp @@ -101,6 +101,10 @@ bool mlirContextIsRegisteredOperation(MlirContext context, MlirStringRef name) { return unwrap(context)->isOperationRegistered(unwrap(name)); } +bool mlirContextIsMultithreadingEnabled(MlirContext context) { + return unwrap(context)->isMultithreadingEnabled(); +} + void mlirContextEnableMultithreading(MlirContext context, bool enable) { return unwrap(context)->enableMultithreading(enable); } |