Skip to content

Commit dfa17f2

Browse files
jacob-shopstekknolagi
authored andcommitted
Fix graphviz generation
1 parent ff55280 commit dfa17f2

1 file changed

Lines changed: 13 additions & 8 deletions

File tree

zjit/src/hir_effect/gen_hir_effect.rb

Lines changed: 13 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -24,20 +24,20 @@ def subeffect name
2424
end
2525

2626
# Helper to generate graphviz.
27-
def to_graphviz_rec effect
27+
def to_graphviz_rec effect, f
2828
effect.subeffects.each {|subeffect|
29-
puts effect.name + "->" + subeffect.name + ";"
29+
f.puts effect.name + "->" + subeffect.name + ";"
3030
}
31-
effect.subeffect.each {|subeffect|
32-
to_graphviz_rec subeffect
31+
effect.subeffects.each {|subeffect|
32+
to_graphviz_rec subeffect, f
3333
}
3434
end
3535

3636
# Generate graphviz.
37-
def to_graphviz effect
38-
puts "digraph G {"
39-
to_graphviz_rec effect
40-
puts "}"
37+
def to_graphviz effect, f
38+
f.puts "digraph G {"
39+
to_graphviz_rec effect, f
40+
f.puts "}"
4141
end
4242

4343
# ===== Start generating the effect DAG =====
@@ -117,3 +117,8 @@ def to_graphviz effect
117117
puts " pub const #{effect_name}: Effect = Effect::promote(abstract_heaps::#{effect_name});"
118118
}
119119
puts "}"
120+
121+
File.open("zjit_effects.dot", "w") do |f|
122+
to_graphviz(any, f)
123+
end
124+

0 commit comments

Comments
 (0)