}
}
-/* Return the single child branch of this predicate.
- * Return NULL if this predicate has no children.
+/* Follow the child if any child whose FuncInst matches with inst
+ *
+ * @param inst must be an ATOMIC_WRITE FuncInst
+ * @return NULL if no such child is found.
*/
-Predicate * Predicate::get_single_child(FuncInst * inst)
+Predicate * Predicate::follow_write_child(FuncInst * inst)
{
- int size = children.size();
- if (size == 0)
- return NULL;
+ ASSERT(inst->get_type() == ATOMIC_WRITE);
- /* Should only have one child */
- ASSERT(size == 1);
- Predicate * child = children[0];
-
- ASSERT(child->get_func_inst() == inst);
+ for (uint i = 0; i < children.size(); i++) {
+ Predicate * child = children[i];
+ if (child->get_func_inst() == inst)
+ return child;
+ }
- return child;
+ return NULL;
}
/* Evaluate predicate expressions against the given inst_act_map */