Forward diagnostics from untracked threads in ParallelDiagnosticHandler.

This allows for the use of multiple ParallelDiagnosticHandlers without having them conflict with each other.

PiperOrigin-RevId: 268967407

1 files changed, 6 insertions, 0 deletions

diff --git a/mlir/lib/Pass/Pass.cpp b/mlir/lib/Pass/Pass.cpp
index ee5ad2e9166..5d5ed9d78a8 100644
--- a/mlir/lib/Pass/Pass.cpp
+++ b/mlir/lib/Pass/Pass.cpp
@@ -410,6 +410,12 @@ void OpToOpPassAdaptorParallel::runOnOperation() {
           if (instrumentor)
             instrumentor->runAfterPipeline(pm->getOpName(), parentThreadID);
 
+          // Drop this thread from being tracked by the diagnostic handler.
+          // After this task has finished, the thread may be used outside of
+          // this pass manager context meaning that we don't want to track
+          // diagnostics from it anymore.
+          diagHandler.eraseOrderIDForThread();
+
           // Handle a failed pipeline result.
           if (failed(pipelineResult)) {
             passFailed = true;