Skip to content

Print ZJIT time stats at the end #387

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Aug 22, 2025
Merged

Print ZJIT time stats at the end #387

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
35 changes: 26 additions & 9 deletions harness/harness-common.rb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -127,8 +127,9 @@ def return_results(warmup_iterations, bench_iterations)
"bench" => bench_iterations,
}

# Collect yjit stats before loading any additional code.
# Collect JIT stats before loading any additional code.
yjit_stats = RubyVM::YJIT.runtime_stats if defined?(RubyVM::YJIT) && RubyVM::YJIT.enabled?
zjit_stats = RubyVM::ZJIT.stats if defined?(RubyVM::ZJIT) && RubyVM::ZJIT.enabled?

# Collect our own peak mem usage as soon as reasonable after finishing the last iteration.
rss = get_rss
Expand All @@ -137,24 +138,40 @@ def return_results(warmup_iterations, bench_iterations)
yjit_bench_results["maxrss"] = maxrss
end

# If YJIT or ZJIT is enabled, show some of their stats at the end.
if yjit_stats
yjit_bench_results["yjit_stats"] = yjit_stats

formatted_stats = proc { |key| "%10s" % format_number(yjit_stats[key]) }
yjit_stats_keys = [
stats_keys = [
*ENV.fetch("YJIT_BENCH_STATS", "").split(",").map(&:to_sym),
:inline_code_size,
:outlined_code_size,
:code_region_size,
:yjit_alloc_size,
:compile_time_ns,
].uniq
yjit_stats_pads = yjit_stats_keys.map(&:size).max + 1
yjit_stats_keys.each do |key|
if key == :compile_time_ns
puts "#{"yjit_compile_time:".ljust(yjit_stats_pads)} %8.2fms" % (yjit_stats[:compile_time_ns] / 1_000_000.0).round(2)
puts "YJIT stats:"
elsif zjit_stats
yjit_bench_results["zjit_stats"] = zjit_stats
stats_keys = [
:compile_time_ns,
:profile_time_ns,
:gc_time_ns,
:invalidation_time_ns,
].uniq
puts "ZJIT stats:"
end
if stats_keys
jit_stats = yjit_stats || zjit_stats
formatted_stats = proc { |key| "%11s" % format_number(jit_stats[key]) }

stats_pads = stats_keys.map(&:size).max + 1
stats_keys.each do |key|
case key
when /_time_ns\z/
key_name = key.to_s.sub(/_time_ns\z/, '_time')
puts "#{key_name.ljust(stats_pads)} %9.2fms" % (jit_stats[key] / 1_000_000.0).round(2)
else
puts "#{"#{key}:".ljust(yjit_stats_pads)} #{formatted_stats[key]}"
puts "#{"#{key}:".ljust(stats_pads)} #{formatted_stats[key]}"
end
end
end
Expand Down