Visualizing MLIR codes to generate DFGs through graphviz

test/visualize/README.md

@@ -0,0 +1,19 @@
+# MLIR Code Visualization Guide
+
+This guide explains how to visualize MLIR (Multi-Level Intermediate Representation) code using various tools and techniques.
+
+## Usage
+
+First, run `apt install graphviz` to install Graphviz. For an MLIR file, use the option `--view-op-graph` to generate a Graphviz visualization of a function. Then, use the `dot` command to create an image file from the visualization.
+
+Example:
+
+```bash
+mlir-neura-opt --assign-accelerator --lower-arith-to-neura --fuse-patterns test.mlir > test_fused.mlir
+mlir-opt --allow-unregistered-dialect --view-op-graph test_fused.mlir 2> test.dot 
+dot -Tpng test.dot -o test.png
+```
+
+It will generate a `test.png` file in the current directory, as shown below:
+
+![test](test.png)

test/visualize/test.mlir

@@ -0,0 +1,13 @@
+// Test cases for FusePatternsPass
+
+// Test 1: Verify dot generation for fused operations
+// RUN: mlir-neura-opt --assign-accelerator --lower-arith-to-neura --fuse-patterns --generate-dot %s | FileCheck %s --check-prefix=CHECK-DOT-GENERATION
+
+func.func @test_dot_generation(%a: f32, %b: f32, %c: f32) -> f32 {
+  %temp = arith.addf %a, %b : f32
+  %res = arith.addf %temp, %c : f32
+  // CHECK-DOT-GENERATION: // DOT: digraph
+  // CHECK-DOT-GENERATION: // DOT: "fadd_fadd"
+  return %res : f32
+}
+

test/visualize/visualize.sh

@@ -0,0 +1,3 @@
+mlir-neura-opt --assign-accelerator --lower-arith-to-neura --fuse-patterns test.mlir > test_fused.mlir
+mlir-opt --allow-unregistered-dialect --view-op-graph test_fused.mlir 2> test.dot 
+dot -Tpng test.dot -o test.png