Unverified Commit 0f8ea0eb authored by Scott Cyphers's avatar Scott Cyphers Committed by GitHub

Cyphers/25convprov (#3655)

* OSX SDK links changed

* Add provenance for ops created in op constructors
Fix optimization in provenance groups
parent 3ae49376
......@@ -339,14 +339,18 @@ shared_ptr<Node> Node::add_provenance_group_members_above(const OutputVector& ba
{
Node* node = todo.back();
todo.pop_back();
if (base_set.count(node) > 0 || !node->m_provenance_group.empty())
if (base_set.count(node) > 0)
{
continue;
}
add_provenance_group_member(node->shared_from_this());
for (auto input : node->inputs())
{
todo.push_back(input.get_source_output().get_node());
if (0 ==
node->m_provenance_group.count(input.get_source_output().get_node_shared_ptr()))
{
todo.push_back(input.get_source_output().get_node());
}
}
base_set.insert(node);
}
......
......@@ -31,6 +31,7 @@ op::util::ArithmeticReduction::ArithmeticReduction(const Output<Node>& arg,
element::i64, Shape{reduction_axes.size()}, reduction_axes.to_vector())
->output(0)})
{
add_provenance_group_member(input(1).get_source_output().get_node_shared_ptr());
}
op::util::ArithmeticReduction::ArithmeticReduction(const Output<Node>& arg,
......
......@@ -30,6 +30,7 @@ op::util::LogicalReduction::LogicalReduction(const Output<Node>& arg, const Axis
element::i64, Shape{reduction_axes.size()}, reduction_axes.to_vector())
->output(0)})
{
add_provenance_group_member(input(1).get_source_output().get_node_shared_ptr());
}
op::util::LogicalReduction::LogicalReduction(const Output<Node>& arg,
......
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