avoid key shuffling with networkx.Graph

coneoproject · Dec 12, 2016 · 9f5050b · 9f5050b
1 parent 955d29d
commit 9f5050b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/coffee/rewriter.py b/coffee/rewriter.py
@@ -568,7 +568,7 @@ def sharing_graph_rewrite(self):
             self.licm('only_const').licm('only_outlinear')
 
         # Transform the expression based on the sharing graph
-        nodes, edges = sgraph.nodes(), sgraph.edges()
+        nodes = [n for n in nodes if n in sgraph.nodes()]
         if not (nodes and all(sgraph.degree(n) > 0 for n in nodes)):
             self.factorize(mode='heuristic')
             self.licm('only_const').licm('only_outlinear')