There is a open project for MLIR visualization, refer to: https://github.com/llvm/llvm-project/blob/main/mlir/lib/Transforms/ViewOpGraph.cpp, Open Projects - MLIR .
But there are some shortcoming about graphviz.
- graphviz does not support subgraph folding. Nesting structre is one of the primary feature of MLIR. Without folding makes it inconvenient to view graph.
- graph dumped by ViewOpGraph does not contain attributes of op.
- It’s hard to(some time timeout) to load big MLIR graph in graphviz.
It will be pefect if we got a tool like TensorBoard | TensorFlow , the function for viewing the model graph.