Updated benchmark scripts

gettalong · Mar 13, 2022 · bde0a90 · bde0a90
1 parent 7f946ce
commit bde0a90
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/benchmark/benchmark-rubies.sh b/benchmark/benchmark-rubies.sh
@@ -6,7 +6,7 @@ eval $(grep -e '^TMPDIR' benchmark/benchmark.sh)
 cd $TMPDIR
 
 for I in kramdown-ruby*.dat; do
-  echo $I | sed -E 's/.*(ruby-.*p[0-9]+(-jit)?).*/\1 ||/' > $I.ruby
+  echo $I | sed -E 's/.*(ruby-.*p[0-9]+(-[my]jit)?).*/\1 ||/' > $I.ruby
   awk '{ print $2 }' $I | tail -n +2 >> $I.ruby
 done
 paste <(awk '{ print $1 }' $(ls kramdown-ruby*.dat | head -n 1)) *.ruby > kramdown-rubies.dat

diff --git a/benchmark/benchmark.sh b/benchmark/benchmark.sh
@@ -48,9 +48,12 @@ git clone .git ${TMPDIR}/kramdown
 cd ${TMPDIR}/kramdown
 
 for RUBY_VERSION in $RUBY_VERSIONS; do
-  if [[ ${RUBY_VERSION: -1} = j ]]; then
-    rbenv shell ${RUBY_VERSION%j}
-    export RUBYOPT=--jit
+  if [[ ${RUBY_VERSION: -1} = m ]]; then
+    rbenv shell ${RUBY_VERSION%m}
+    export RUBYOPT=--mjit
+  elif [[ ${RUBY_VERSION: -1} = y ]]; then
+    rbenv shell ${RUBY_VERSION%y}
+    export RUBYOPT=--yjit
   else
     rbenv shell $RUBY_VERSION
     unset RUBYOPT

diff --git a/benchmark/generate_data.rb b/benchmark/generate_data.rb
@@ -14,7 +14,8 @@
 
 THISRUBY = (self.class.const_defined?(:RUBY_DESCRIPTION) ? RUBY_DESCRIPTION.scan(/^.*?(?=\s*\(|,)/).first.sub(/\s/, '-') : "ruby-#{RUBY_VERSION}")
 THISRUBY << '-' + RUBY_PATCHLEVEL.to_s unless THISRUBY =~ /p#{RUBY_PATCHLEVEL}$/
-THISRUBY << '-jit' if RUBY_DESCRIPTION =~ /JIT/
+THISRUBY << '-mjit' if RUBY_DESCRIPTION =~ /MJIT/
+THISRUBY << '-yjit' if RUBY_DESCRIPTION =~ /YJIT/
 
 Dir.chdir(File.dirname(__FILE__))
 BMDATA = File.read('mdbasics.text')