[Mlir-commits] [mlir] Fix crash in --test-tosa-op-availability when running under the threaded pass manager (PR #183022)

[llvmlistbot at llvm.org]

llvmbot wrote:


<!--LLVM PR SUMMARY COMMENT-->

@llvm/pr-subscribers-mlir

Author: Ayush Kumar Gaur (Ayush3941)

<details>
<summary>Changes</summary>

whats this Pr does :-
The PR adds a static lock (or lock + buffering) so only one thread prints at once, preventing raw_ostream assertions and garbled output.

---
Full diff: https://github.com/llvm/llvm-project/pull/183022.diff


1 Files Affected:

- (modified) mlir/test/lib/Dialect/Tosa/TestAvailability.cpp (+3) 


``````````diff
diff --git a/mlir/test/lib/Dialect/Tosa/TestAvailability.cpp b/mlir/test/lib/Dialect/Tosa/TestAvailability.cpp
index bec563d1ec747..9bd13a4518457 100644
--- a/mlir/test/lib/Dialect/Tosa/TestAvailability.cpp
+++ b/mlir/test/lib/Dialect/Tosa/TestAvailability.cpp
@@ -26,9 +26,12 @@ struct PrintOpAvailability
   StringRef getArgument() const final { return "test-tosa-op-availability"; }
   StringRef getDescription() const final { return "Test Tosa op availability"; }
 };
+static llvm::sys::SmartRWMutex<true> PrintMutex;
 } // namespace
 
 void PrintOpAvailability::runOnOperation() {
+  llvm::sys::SmartScopedWriter<true> lock(PrintMutex);
+
   auto f = getOperation();
   llvm::outs() << f.getName() << "\n";
 

``````````

</details>


https://github.com/llvm/llvm-project/pull/183022