add extra debug statements

j2kun · Nov 14, 2023 · b78a997 · b78a997
1 parent e0b8017
commit b78a997
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/lib/Transform/Noisy/ReduceNoiseOptimizer.cpp b/lib/Transform/Noisy/ReduceNoiseOptimizer.cpp
@@ -8,11 +8,14 @@
 #include "mlir/include/mlir/Analysis/DataFlowFramework.h"
 #include "mlir/include/mlir/IR/Visitors.h"
 #include "mlir/include/mlir/Pass/Pass.h"
+#include "llvm/Support/Debug.h"
 
 namespace mlir {
 namespace tutorial {
 namespace noisy {
 
+#define DEBUG_TYPE "ReduceNoiseOptimizer"
+
 #define GEN_PASS_DEF_REDUCENOISEOPTIMIZER
 #include "lib/Transform/Noisy/Passes.h.inc"
 
@@ -43,8 +46,11 @@ struct ReduceNoiseOptimizer
     // dependence is not automatic and fails silently.
     solver.load<dataflow::DeadCodeAnalysis>();
     solver.load<dataflow::IntegerRangeAnalysis>();
-    if (failed(solver.initializeAndRun(module)))
+    if (failed(solver.initializeAndRun(module))) {
+      getOperation()->emitOpError() << "Failed to run the analysis.\n";
       signalPassFailure();
+      return;
+    }
 
     auto result = module->walk([&](Operation *op) {
       if (!llvm::isa<noisy::AddOp, noisy::SubOp, noisy::MulOp,
@@ -61,6 +67,9 @@ struct ReduceNoiseOptimizer
       }
 
       ConstantIntRanges range = opRange->getValue().getValue();
+      LLVM_DEBUG(llvm::dbgs() << "op " << *op << " has error range ["
+                              << range.umin().getZExtValue() << ", "
+                              << range.umax().getZExtValue() << "]\n");
       if (range.umax().getZExtValue() > MAX_NOISE) {
         op->emitOpError() << "Found op after which the noise exceeds the "
                              "allowable maximum of "
@@ -73,8 +82,11 @@ struct ReduceNoiseOptimizer
       return WalkResult::advance();
     });
 
-    if (result.wasInterrupted())
+    if (result.wasInterrupted()) {
+      getOperation()->emitOpError()
+          << "Detected error in the noise analysis.\n";
       signalPassFailure();
+    }
   }
 };