Commit 87348cab authored by Nick Korovaiko's avatar Nick Korovaiko Committed by Robert Kimball

make sure visualize_tree always uses a details map (#1744)

parent 58df83cf
...@@ -90,6 +90,10 @@ void ngraph::pass::Manager::run_passes(shared_ptr<Function> func, bool transitiv ...@@ -90,6 +90,10 @@ void ngraph::pass::Manager::run_passes(shared_ptr<Function> func, bool transitiv
auto call_graph_pass = dynamic_pointer_cast<CallGraphPass>(pass); auto call_graph_pass = dynamic_pointer_cast<CallGraphPass>(pass);
if (module_pass) if (module_pass)
{ {
if (auto vt_pass = dynamic_pointer_cast<pass::VisualizeTree>(module_pass))
{
vt_pass->set_ops_to_details(get_state().get_visualize_tree_ops_map());
}
module_pass->run_on_module(fs); module_pass->run_on_module(fs);
} }
else if (function_pass) else if (function_pass)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment