summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--test/ruby/test_zjit.rb12
-rw-r--r--zjit.rb6
2 files changed, 15 insertions, 3 deletions
diff --git a/test/ruby/test_zjit.rb b/test/ruby/test_zjit.rb
index 333e34e927..ff7c167327 100644
--- a/test/ruby/test_zjit.rb
+++ b/test/ruby/test_zjit.rb
@@ -27,6 +27,18 @@ class TestZJIT < Test::Unit::TestCase
RUBY
end
+ def test_stats_string_no_zjit
+ assert_runs 'nil', <<~RUBY, zjit: false
+ RubyVM::ZJIT.stats_string
+ RUBY
+ assert_runs 'true', <<~RUBY, stats: false
+ RubyVM::ZJIT.stats_string.is_a?(String)
+ RUBY
+ assert_runs 'true', <<~RUBY, stats: true
+ RubyVM::ZJIT.stats_string.is_a?(String)
+ RUBY
+ end
+
def test_stats_quiet
# Test that --zjit-stats-quiet collects stats but doesn't print them
script = <<~RUBY
diff --git a/zjit.rb b/zjit.rb
index 0cc9ca8261..f73ef12a63 100644
--- a/zjit.rb
+++ b/zjit.rb
@@ -165,13 +165,13 @@ class << RubyVM::ZJIT
# Get the summary of ZJIT statistics as a String
def stats_string
+ return unless stats = self.stats
buf = +"***ZJIT: Printing ZJIT statistics on exit***\n"
- stats = self.stats
- if stats[:guard_type_count].nonzero?
+ if stats[:guard_type_count]&.nonzero?
stats[:guard_type_exit_ratio] = stats[:exit_guard_type_failure].to_f / stats[:guard_type_count] * 100
end
- if stats[:guard_shape_count].nonzero?
+ if stats[:guard_shape_count]&.nonzero?
stats[:guard_shape_exit_ratio] = stats[:exit_guard_shape_failure].to_f / stats[:guard_shape_count] * 100
end