add option view-op-graph in mlir-neura-opt

Author: jiajun12138

test/visualize/README.md

@@ -9,8 +9,7 @@ First, run `apt install graphviz` to install Graphviz. For an MLIR file, use the
 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 
+mlir-neura-opt --assign-accelerator --lower-arith-to-neura --fuse-pattern --view-op-graph test.mlir 2> test.dot 
 dot -Tpng test.dot -o test.png
 ```
 

test/visualize/test.mlir

@@ -1,13 +1,12 @@
-// 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
+// RUN: mlir-neura-opt --assign-accelerator --lower-arith-to-neura --fuse-pattern --view-op-graph %s 2>&1 | 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"
+  // CHECK-DOT-GENERATION: digraph G
+  // CHECK-DOT-GENERATION: label = "func.func : ()\n\naccelerator: \"neura\"\nfunction_type: (f32, f32, f32) -> f...\nsym_name: \"test_dot_generation...";
+  // CHECK-DOT-GENERATION: label = "neura.fadd_fadd : (f32)\n"
   return %res : f32
 }
 

tools/mlir-neura-opt/mlir-neura-opt.cpp

@@ -24,6 +24,7 @@ int main(int argc, char **argv) {
 
   mlir::neura::registerPasses();
   mlir::registerPasses();
+  mlir::registerViewOpGraphPass();
 
   // Runs the MLIR optimizer.
   return mlir::asMainReturnCode(