make more benchmarks ractor friendly

Author: luke-gruber

benchmarks.yml

@@ -60,6 +60,7 @@ etanni:
   desc: etanni is an older, extremely simple template-lang format that basically turns your template into an "eval" with a lot of heredocs.
 fannkuchredux:
   desc: fannkuchredux from the Computer Language Benchmarks Game.
+  ractor: true
 fluentd:
   desc: fluentd is a log collector, which parses logs in a server and forwards them to various destinations.
 graphql:
@@ -81,8 +82,10 @@ ruby-json:
   ractor: true
 rubyboy:
   desc: Rubyboy is a functional headless GameBoy emulator, run on a specific game cartridge for a specific number of frames.
+  ractor: true
 rubykon:
   desc: Ruby solver for Go (the boardgame.) Runs many iterations forward from an initial starting board.
+  ractor: true
 tinygql:
   desc: TinyGQL gem parsing a large file in pure Ruby
 nqueens:
@@ -99,8 +102,10 @@ blurhash:
   ractor: true
 protoboeuf:
   desc: protoboeuf (pure-Ruby protobuf) message decoding
+  ractor: true
 protoboeuf-encode:
   desc: protoboeuf (pure-Ruby protobuf) message encoding
+  ractor: true
 
 #
 # MicroBenchmarks

benchmarks/fannkuchredux/benchmark.rb

@@ -54,12 +54,12 @@ def fannkuch(n)
 end
 
 #n = (ARGV[0] || 1).to_i
-n = 9 # Benchmarks Game uses n = 12, but it's too slow
+N = 9 # Benchmarks Game uses n = 12, but it's too slow
 
 require_relative '../../harness/loader'
 
 run_benchmark(10) do
-  sum, flips = fannkuch(n)
+  sum, flips = fannkuch(N)
 
   if sum != 8629
     raise RuntimeError, "incorrect sum: #{sum}"

benchmarks/protoboeuf-encode/benchmark.rb

@@ -5,8 +5,8 @@
 
 Dir.chdir __dir__
 fake_msg_bins = Marshal.load(File.binread('encoded_msgs.bin'))
-lots = fake_msg_bins.map { |bin| ProtoBoeuf::ParkingLot.decode bin }
+LOTS = Ractor.make_shareable(fake_msg_bins.map { |bin| ProtoBoeuf::ParkingLot.decode bin })
 
 run_benchmark(20) do
-  lots.each { |lot| ProtoBoeuf::ParkingLot.encode lot }
+  LOTS.each { |lot| ProtoBoeuf::ParkingLot.encode lot }
 end

benchmarks/protoboeuf/benchmark.rb

@@ -4,8 +4,8 @@
 require_relative 'benchmark_pb'
 
 Dir.chdir __dir__
-fake_msg_bins = Marshal.load(File.binread('encoded_msgs.bin'))
+FAKE_MSG_BINS = Ractor.make_shareable(Marshal.load(File.binread('encoded_msgs.bin')))
 
 run_benchmark(20) do
-  fake_msg_bins.each { |bin| ProtoBoeuf::ParkingLot.decode bin }
+  FAKE_MSG_BINS.each { |bin| ProtoBoeuf::ParkingLot.decode bin }
 end

benchmarks/rubyboy/benchmark.rb

@@ -11,18 +11,21 @@
 require 'rubyboy/emulator_headless'
 
 # The rom is included in the gem in a sibling directory to the rubyboy code.
-rom_path = File.expand_path("../../roms/tobu.gb", $".detect { |x| x.end_with?("/rubyboy/emulator_headless.rb") })
+ROM_PATH = File.expand_path("../../roms/tobu.gb", $".detect { |x| x.end_with?("/rubyboy/emulator_headless.rb") }).freeze
 
 # A count of 500 produces results similar to our optcarrot benchmark.
 # It's possible there is a number that produces a consistent benchmark without
 # needing to re-initialize but not sure how to determine that.
-count = 500
+COUNT = 500
+
+Ractor.make_shareable(Rubyboy::ApuChannels::Channel1::WAVE_DUTY)
+Ractor.make_shareable(Rubyboy::ApuChannels::Channel2::WAVE_DUTY)
 
 run_benchmark(200) do
   # Results are much more consistent if we re-initialize each time.
   # Reusing the same eumlator increases stddev by 65x.
-  emulator = Rubyboy::EmulatorHeadless.new(rom_path)
-  count.times do
+  emulator = Rubyboy::EmulatorHeadless.new(ROM_PATH)
+  COUNT.times do
     emulator.step
   end
 end

benchmarks/rubykon/benchmark.rb

@@ -10,9 +10,17 @@
 # return the same stable best_move, even for identical initial board state and number of iterations.
 
 ITERATIONS = 100
-game_state = Rubykon::GameState.new Rubykon::Game.new(19)
-mcts = MCTS::MCTS.new
 
-run_benchmark(10) do
-  mcts.start game_state, ITERATIONS
+if ENV["YJIT_BENCH_RACTOR_HARNESS"]
+  run_benchmark(10) do
+    state = Rubykon::GameState.new Rubykon::Game.new(19)
+    m = MCTS::MCTS.new
+    m.start state, ITERATIONS
+  end
+else
+  game_state = Rubykon::GameState.new Rubykon::Game.new(19)
+  mcts = MCTS::MCTS.new
+  run_benchmark(10) do
+    mcts.start game_state, ITERATIONS
+  end
 end