From f2fb41cf1f8fb1cc58238929e1821c3b50eb6f77 Mon Sep 17 00:00:00 2001 From: FXTi Date: Tue, 27 Aug 2019 17:22:43 +0800 Subject: [PATCH] Fix Entry node missing problem happening in v2 version --- simplifier.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/simplifier.py b/simplifier.py index d02eeb9..21207b7 100644 --- a/simplifier.py +++ b/simplifier.py @@ -50,6 +50,9 @@ def eliminate_forwarders(self): if forwardedBB_in_edge_exists and forwarderBB_in_edge_exists: continue + if forwarderBB in nx.get_node_attributes(self.bb_graph,'isEntry').keys(): + nx.set_node_attributes(self.bb_graph, 'isEntry', {forwardedBB: True}) + # Remove the edge between forwarder and forwarded self.bb_graph.remove_edge(forwarderBB, forwardedBB)