1 files changed, 56 insertions, 1 deletions
diff --git a/mlir/lib/Transforms/GenerateRuntimeVerification.cpp b/mlir/lib/Transforms/GenerateRuntimeVerification.cpp
index a40bc2b..63c71cd 100644
--- a/mlir/lib/Transforms/GenerateRuntimeVerification.cpp
+++ b/mlir/lib/Transforms/GenerateRuntimeVerification.cpp
@@ -6,6 +6,7 @@
 //
 //===----------------------------------------------------------------------===//
 
+#include "mlir/IR/AsmState.h"
 #include "mlir/Transforms/Passes.h"
 
 #include "mlir/IR/Builders.h"
@@ -25,9 +26,51 @@ struct GenerateRuntimeVerificationPass
           GenerateRuntimeVerificationPass> {
   void runOnOperation() override;
 };
+
+/// Default error message generator for runtime verification failures.
+///
+/// This class generates error messages with different levels of verbosity:
+/// - Level 0: Shows only the error message and operation location
+/// - Level 1: Shows the full operation string, error message, and location
+///
+/// Clients can call getVerboseLevel() to retrieve the current verbose level
+/// and use it to customize their own error message generators with similar
+/// behavior patterns.
+class DefaultErrMsgGenerator {
+private:
+  unsigned vLevel;
+  AsmState &state;
+
+public:
+  DefaultErrMsgGenerator(unsigned verboseLevel, AsmState &asmState)
+      : vLevel(verboseLevel), state(asmState) {}
+
+  std::string operator()(Operation *op, StringRef msg) {
+    std::string buffer;
+    llvm::raw_string_ostream stream(buffer);
+    stream << "ERROR: Runtime op verification failed\n";
+    if (vLevel == 1) {
+      op->print(stream, state);
+      stream << "\n";
+    }
+    stream << "^\nLocation: ";
+    op->getLoc().print(stream);
+    return buffer;
+  }
+
+  unsigned getVerboseLevel() const { return vLevel; }
+};
 } // namespace
 
 void GenerateRuntimeVerificationPass::runOnOperation() {
+  // Check verboseLevel is in range [0, 1].
+  if (verboseLevel > 1) {
+    getOperation()->emitError(
+        "generate-runtime-verification pass: set verboseLevel to 0 or 1");
+    signalPassFailure();
+    return;
+  }
+
   // The implementation of the RuntimeVerifiableOpInterface may create ops that
   // can be verified. We don't want to generate verification for IR that
   // performs verification, so gather all runtime-verifiable ops first.
@@ -36,10 +79,22 @@ void GenerateRuntimeVerificationPass::runOnOperation() {
     ops.push_back(verifiableOp);
   });
 
+  // We may generate a lot of error messages and so we need to ensure the
+  // printing is fast.
+  OpPrintingFlags flags;
+  flags.elideLargeElementsAttrs();
+  flags.skipRegions();
+  flags.useLocalScope();
+  AsmState state(getOperation(), flags);
+
+  // Client can call getVerboseLevel() to fetch verbose level.
+  DefaultErrMsgGenerator defaultErrMsgGenerator(verboseLevel.getValue(), state);
+
   OpBuilder builder(getOperation()->getContext());
   for (RuntimeVerifiableOpInterface verifiableOp : ops) {
     builder.setInsertionPoint(verifiableOp);
-    verifiableOp.generateRuntimeVerification(builder, verifiableOp.getLoc());
+    verifiableOp.generateRuntimeVerification(builder, verifiableOp.getLoc(),
+                                             defaultErrMsgGenerator);
   };
 }