Skip to content

Commit 3849c13

Browse files
committed
perf: use copy=False when reversing graphs
1 parent 3ea3a01 commit 3849c13

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

count_reachable_roots.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def count_reachable_roots(
4141
target_node = files.index(target)
4242

4343
reachable_nodes = set(
44-
[files[idx] for idx in nx.dfs_postorder_nodes(DG.reverse(), source=target_node) if idx != target_node]
44+
[files[idx] for idx in nx.dfs_postorder_nodes(DG.reverse(False), source=target_node) if idx != target_node]
4545
)
4646

4747
reachable_roots = set(include_analysis["roots"]).intersection(reachable_nodes)

cut_header.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ def create_include_graph(
107107
DG[edge[0]][edge[1]]["capacity"] = float("inf")
108108

109109
# Remove everything in the graph that's not reachable from the target to speed up analysis
110-
reachable_nodes = set(nx.dfs_postorder_nodes(DG.reverse(), source=files.index(target)))
110+
reachable_nodes = set(nx.dfs_postorder_nodes(DG.reverse(False), source=files.index(target)))
111111
DG.remove_nodes_from([idx for idx in list(DG.nodes()) if idx not in reachable_nodes])
112112

113113
return DG
@@ -238,7 +238,7 @@ def compute_top_indirect_cuts(
238238
DG2.add_node(PSEUDO_SOURCE)
239239

240240
# Find reachable roots
241-
reachable_nodes = set(files[idx] for idx in nx.dfs_postorder_nodes(DG2.reverse(), source=target_idx))
241+
reachable_nodes = set(files[idx] for idx in nx.dfs_postorder_nodes(DG2.reverse(False), source=target_idx))
242242
reachable_roots = [
243243
root
244244
for root in include_analysis["roots"]
@@ -445,7 +445,7 @@ def compute_doms_to_target(include_analysis: IncludeAnalysisOutput, DG: nx.DiGra
445445
target_idx = files.index(target)
446446

447447
# Find reachable roots
448-
reachable_nodes = set(files[idx] for idx in nx.dfs_postorder_nodes(DG2.reverse(), source=target_idx))
448+
reachable_nodes = set(files[idx] for idx in nx.dfs_postorder_nodes(DG2.reverse(False), source=target_idx))
449449
roots = [
450450
root
451451
for root in include_analysis["roots"]

0 commit comments

Comments
 (0)