Add support for MLIR_ENABLE_REMARK (#2625)

Signed-off-by: Tiotto, Ettore <[email protected]>

Author: etiotto

third_party/intel/backend/compiler.py

@@ -210,6 +210,11 @@ def make_ttgir(mod, metadata, opt, properties):
             cluster_info.clusterDimX = opt.cluster_dims[0]
             cluster_info.clusterDimY = opt.cluster_dims[1]
             cluster_info.clusterDimZ = opt.cluster_dims[2]
+        # Set up Diagnostic
+        if os.environ.get("MLIR_ENABLE_REMARK", "0") == "1":
+            srcMgr = llvm.source_mgr()
+            ir.source_mgr_diag(srcMgr, mod.context)
+            mod.context.printOpOnDiagnostic(True)
 
         # Annotate module with information required by subsequent transformations.
         pm = ir.pass_manager(mod.context)
@@ -270,6 +275,11 @@ def make_llir(src, metadata, options):
         # TritonGPU -> LLVM-IR (MLIR)
         pm = ir.pass_manager(mod.context)
         pm.enable_debug()
+        # Set up Diagnostic
+        if os.environ.get("MLIR_ENABLE_REMARK", "0") == "1":
+            srcMgr = llvm.source_mgr()
+            ir.source_mgr_diag(srcMgr, mod.context)
+            mod.context.printOpOnDiagnostic(True)
         # FIXME: Advanced path drops tensor layouts, so this will crash on some
         # operations being used, e.g., convert_layout.
         if os.getenv("TRITON_INTEL_REDUCE_TRANSPOSE", "0") != "1":