diff --git a/paddle/fluid/inference/utils/visualizer.cc b/paddle/fluid/inference/utils/visualizer.cc index 7c0dd64dea88e5..6bada552d80f62 100644 --- a/paddle/fluid/inference/utils/visualizer.cc +++ b/paddle/fluid/inference/utils/visualizer.cc @@ -74,6 +74,7 @@ int main(int argc, char *argv[]) { // Only 1 pass, default filename is 0_ir_origin.dot // For more details, looking for paddle::inference::analysis::IRPassManager argument.SetIrAnalysisPasses({"infer_clean_graph_pass", "graph_viz_pass"}); + argument.SetAnalysisPasses({"ir_graph_build_pass", "ir_analysis_pass"}); std::unique_ptr scope{ new paddle::framework::Scope()};