A new Beginning

src/PIM/Common/Support/DebugDump.cpp

@@ -17,7 +17,7 @@ std::fstream openDialectDumpFileWithExtension(const std::string& name, llvm::Str
   return std::fstream(dialectsDir + "/" + name + "." + extension.str(), std::ios::out);
 }
 
-void dumpModule(mlir::ModuleOp moduleOp, const std::string& name) {
+void dumpModule(mlir::ModuleOp moduleOp, const std::string& name, bool assumeVerified) {
   std::fstream file = openDialectDumpFileWithExtension(name, "/dialects", "mlir");
   if (!file.is_open())
     return;
@@ -25,6 +25,8 @@ void dumpModule(mlir::ModuleOp moduleOp, const std::string& name) {
   llvm::raw_os_ostream os(file);
   mlir::OpPrintingFlags flags;
   flags.elideLargeElementsAttrs().enableDebugInfo(true, false);
+  if (assumeVerified)
+    flags.assumeVerified();
   moduleOp.print(os, flags);
   os.flush();
   file.close();

src/PIM/Common/Support/DebugDump.hpp

@@ -10,7 +10,7 @@ namespace onnx_mlir {
 
 /// Emits a MLIR snapshot under the current compiler output
 /// directory for pass-level debugging.
-void dumpModule(mlir::ModuleOp moduleOp, const std::string& name);
+void dumpModule(mlir::ModuleOp moduleOp, const std::string& name, bool assumeVerified = false);
 
 /// Opens a file under the same dialect dump directory used by dumpModule.
 std::fstream openDialectDumpFileWithExtension(const std::string& name,llvm::StringRef destination = "/dialects",  llvm::StringRef extension = "mlir");