[mlir][core] Restore dropped printIR behavior. (#196628)

cathyzhyi · web-flow · commit 0236ad590897 · 2026-05-08T13:18:52.000-07:00
Restore checking for module scope which is dropped in #195198
diff --git a/mlir/lib/Pass/IRPrinting.cpp b/mlir/lib/Pass/IRPrinting.cpp
@@ -48,6 +48,11 @@ class IRPrinterInstrumentation : public PassInstrumentation {
 
 static void printIR(Operation *op, bool printModuleScope, raw_ostream &out,
                     OpPrintingFlags flags) {
+  // Check to see if we are not printing at module scope.
+  if (!printModuleScope)
+    return op->print(out, op->getBlock() ? flags.useLocalScope() : flags);
+
+  // Otherwise, we are printing at module scope.
   // Find the top-level operation.
   auto *topLevelOp = op;
   while (auto *parentOp = topLevelOp->getParentOp())
