maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[mlir] Destroy MLIRContext thread pool when disable multi threading 

Reviewed By: rriddle

Differential Revision: https://reviews.llvm.org/D105057
This commit is contained in:
Eugene Zhulenev 2021-06-28 13:15:24 -07:00
parent 7ac0442fe5
commit fe3c425ae0
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
mlir/lib/IR/MLIRContext.cpp
View File

@ -262,7 +262,7 @@ public:
//===--------------------------------------------------------------------===//
/// The thread pool to use when processing MLIR tasks in parallel.
llvm::ThreadPool threadPool;
llvm::Optional<llvm::ThreadPool> threadPool;
/// This is a list of dialects that are created referring to this context.
/// The MLIRContext owns the objects.
@ -334,7 +334,10 @@ public:
StringAttr emptyStringAttr;
public:
MLIRContextImpl() : identifiers(identifierAllocator) {}
MLIRContextImpl() : identifiers(identifierAllocator) {
if (threadingIsEnabled)
threadPool.emplace();
}
~MLIRContextImpl() {
for (auto typeMapping : registeredTypes)
typeMapping.second->~AbstractType();
@ -573,12 +576,19 @@ void MLIRContext::disableMultithreading(bool disable) {
impl->affineUniquer.disableMultithreading(disable);
impl->attributeUniquer.disableMultithreading(disable);
impl->typeUniquer.disableMultithreading(disable);
// Destroy thread pool (stop all threads) if it is no longer needed, or create
// a new one if multithreading was re-enabled.
if (!impl->threadingIsEnabled)
impl->threadPool.reset();
else if (!impl->threadPool.hasValue())
impl->threadPool.emplace();
}
llvm::ThreadPool &MLIRContext::getThreadPool() {
assert(isMultithreadingEnabled() &&
"expected multi-threading to be enabled within the context");
return impl->threadPool;
return *impl->threadPool;
}
void MLIRContext::enterMultiThreadedExecution() {