Adding context option to print mlir stactrace on error

tenstorrent · Aug 16, 2024 · 3bcfe05 · 3bcfe05
1 parent 0afb0f4
commit 3bcfe05
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pybuda/csrc/passes/mlir_compiler.cpp b/pybuda/csrc/passes/mlir_compiler.cpp
@@ -42,6 +42,12 @@ namespace tt::passes
 
         // Create a context with all registered dialects.
         mlir::MLIRContext context(registry);
+
+#ifdef DEBUG
+        // Context setting to have mlir print out stacktrace whenever errors occur
+        context.printStackTraceOnDiagnostic(true);
+#endif
+
         // Load all available dialects
         context.loadAllAvailableDialects();