WIP

Author: byroot

benchmarks/binarytrees/benchmark.rb

@@ -5,39 +5,47 @@
 # Modified by Wesley Moxam
 # *reset*
 
-def item_check(left, right)
-  return 1 if left.nil?
-  1 + item_check(*left) + item_check(*right)
-end
+module Btree
+  extend self
 
-def bottom_up_tree(depth)
-  return [nil, nil] unless depth > 0
-  depth -= 1
-  [bottom_up_tree(depth), bottom_up_tree(depth)]
-end
+  def item_check(left, right)
+    return 1 if left.nil?
+    1 + item_check(*left) + item_check(*right)
+  end
 
-max_depth = 14
-min_depth = 4
+  def bottom_up_tree(depth)
+    return [nil, nil] unless depth > 0
+    depth -= 1
+    [bottom_up_tree(depth), bottom_up_tree(depth)]
+  end
 
-max_depth = min_depth + 2 if min_depth + 2 > max_depth
-stretch_depth = max_depth + 1
+  def bench
+    max_depth = 14
+    min_depth = 4
 
-require_relative '../../harness/loader'
+    max_depth = min_depth + 2 if min_depth + 2 > max_depth
+    stretch_depth = max_depth + 1
 
-run_benchmark(60) do
-  stretch_tree = bottom_up_tree(stretch_depth)
-  stretch_tree = nil
+    stretch_tree = bottom_up_tree(stretch_depth)
+    stretch_tree = nil
 
-  long_lived_tree = bottom_up_tree(max_depth)
+    long_lived_tree = bottom_up_tree(max_depth)
 
-  min_depth.step(max_depth, 2) do |depth|
-    iterations = 2**(max_depth - depth + min_depth)
+    min_depth.step(max_depth, 2) do |depth|
+      iterations = 2**(max_depth - depth + min_depth)
 
-    check = 0
+      check = 0
 
-    for i in 1..iterations
-      temp_tree = bottom_up_tree(depth)
-      check += item_check(*temp_tree)
+      for i in 1..iterations
+        temp_tree = bottom_up_tree(depth)
+        check += item_check(*temp_tree)
+      end
     end
   end
 end
+
+require_relative '../../harness/loader'
+
+run_benchmark(60) do
+  Btree.bench
+end

harness/harness.rb

@@ -31,6 +31,20 @@ def realtime
   Process.clock_gettime(Process::CLOCK_MONOTONIC) - r0
 end
 
+def run_once(&block)
+  if parallel = ENV["RACTOR_PARALLEL"]
+    # block = Ractor.make_shareable(block)
+    realtime do
+      ractors = Integer(parallel).times.map do
+        Ractor.new(&block)
+      end
+      ractors.each(&:join)
+    end
+  else
+    realtime(&block)
+  end
+end
+
 # Takes a block as input
 def run_benchmark(_num_itrs_hint, &block)
   times = []
@@ -50,7 +64,7 @@ def run_benchmark(_num_itrs_hint, &block)
   begin
     yjit_stats&.each_key { |key| yjit_stats[key] = RubyVM::YJIT.runtime_stats(key) }
 
-    time = realtime(&block)
+    time = run_once(&block)
     num_itrs += 1
 
     # NOTE: we may want to avoid this as it could trigger GC?