From 4e19650b80b3dd67bf6c5fd791b41f03e64e1724 Mon Sep 17 00:00:00 2001
From: NiccoloN <niccolo.nicolosi@gmail.com>
Date: Wed, 25 Mar 2026 10:11:36 +0100
Subject: [PATCH] minor fix to MessagePass.cpp

---
 src/PIM/Pass/MessagePass.cpp | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/src/PIM/Pass/MessagePass.cpp b/src/PIM/Pass/MessagePass.cpp
index 01224fd..3a897a5 100644
--- a/src/PIM/Pass/MessagePass.cpp
+++ b/src/PIM/Pass/MessagePass.cpp
@@ -1,7 +1,7 @@
 #include "mlir/IR/BuiltinOps.h"
 #include "mlir/Pass/Pass.h"
 
-#include "llvm/Support/raw_ostream.h"
+#include "src/Compiler/CompilerUtils.hpp"
 
 using namespace mlir;
 
@@ -18,10 +18,7 @@ struct MessagePass : PassWrapper<MessagePass, OperationPass<ModuleOp>> {
   : message(message) {}
   MessagePass(const MessagePass& pass) {}
 
-  void runOnOperation() final {
-    llvm::outs() << message << "\n";
-    llvm::outs().flush();
-  }
+  void runOnOperation() final { showCompilePhase(message); }
 
 private:
   std::string message;