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
auto call_graph_pass = dynamic_pointer_cast<CallGraphPass>(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);
}
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