Allow to toggle YJIT stats collection from runtime

For use cases where you want to collect the metrics
for a specific piece of code (typically a web request)
you can have the stats turned off by default and then
turn them on at runtime before executing the code you care
about.

1 files changed, 8 insertions, 0 deletions

diff --git a/yjit.rb b/yjit.rb
index afd1071250..9556f07ad6 100644
--- a/yjit.rb
+++ b/yjit.rb
@@ -134,6 +134,14 @@ module YJIT
     Primitive.reset_stats_bang
   end
 
+  def self.stats_enabled?
+    Primitive.cexpr! 'rb_yjit_opts.gen_stats ? Qtrue : Qfalse'
+  end
+
+  def self.stats_enabled=(enabled)
+    Primitive.set_stats_enabled(enabled)
+  end
+
   def self.enabled?
     Primitive.cexpr! 'rb_yjit_enabled_p() ? Qtrue : Qfalse'
   end