Keyboard shortcuts

Press or to navigate between chapters

Press S or / to search in the book

Press ? to show this help

Press Esc to hide this help

基本演算の再定義(BOP redefinition)

1 + 2 のような演算子式を、メソッド探索を挟まずインライン展開してよいのは 「Integer#+ が再定義されていない」という仮定が成り立つ間だけである。Ruby はその仮定をいつでも壊せる(class Integer; def +(o); …; end; end)ので、 処理系は「仮定を置く」「壊れたことを検知する」「壊れたあと辻褄を合わせる」の 3 点を用意しなければならない。

本書は monoruby の現行実装をコードに即して記述し、CRuby の戦略と実測で比較し、 なぜ今の設計を変えるべきかと、どの順で変えるべきかを記録する。

本書は design record である(doc/README.md の分類)。測定値は 2026-08-10 時点、076cd951、Linux/x86-64、release ビルドのもの。

関連: doc/refinements.md §6.7(refinements と基本演算)、doc/jit.mddoc/inline.md


1. 出発点の実装(§5 Step 1 で置き換え済み)

: 本節は Step 1 以前の実装を記述する。検知まわり(§1.1・§1.2・§1.4)は globals/store/basic_op.rs の静的表に置き換わった(§5 Step 1)。反応 (§1.3)とコスト(§1.5)は現在も同じで、Step 2 の対象である。問題の形を 残すために原文のまま置く。

1.1 「basic op」として登録されていたもの

登録の入口は 1 つだけで、Globals::define_basic_opglobals/method.rs)→ new_basic_opadd_basic_op_methodadd_method_inner(.., is_basic_op = true, ..) と流れ、 MethodTableEntry::is_basic_op を立てる。

呼び出し箇所は全部で 10 個:

クラス演算子
Integer+ - * / !=
Float+ - * !=
String!=

これが「再定義を検知できる集合」の全てである。VM / JIT がインライン展開して いる演算子はこれよりはるかに多い(§1.4)。

1.2 検知

ClassInfoTable::insert_methodglobals/store/class.rs)は、メソッド表を 上書きしたとき 元のエントリが is_basic_op だったかだけを見る:

#![allow(unused)]
fn main() {
fn insert_method(&mut self, class_id: ClassId, name: IdentId, entry: MethodTableEntry) {
    Globals::class_version_inc();
    if let Some(old) = self.classes[class_id].methods.insert(name, entry)
        && old.is_basic_op
    {
        self.set_bop_redefine();
    }
}
}

どの演算子が・どのクラスで再定義されたかは記録されない。 呼ばれた事実だけが 伝わる。

1.3 反応 — グローバルかつ恒久

Store::set_bop_redefine(同ファイル)が一度に次を行う:

  1. Codegen::set_bop_redefinebop_redefined_flags: u32!0 を書く。 以後クリアされない。
  2. Store::invalidate_jit_code()全 iseqjit_invalidated = true。 これも一方向ラッチで、以後その iseq は method-JIT の対象から外れる (compiler.rs / patch.rsjit_invalidated() で早期 return)。
  3. x86-64: 全コンパイル済みメソッドの entry を apply_jmp_patch_addressvm_entry に書き戻す。aarch64: jit_slot / jit_guard_free_slot の dispatch word をゼロ化(invalidate_jit_code 内)。
  4. remove_vm_bop_optimization()VM のディスパッチ表を _no_opt 版に 恒久的に差し替える。算術・比較・単項の fixnum fast path が全て フルディスパッチになる。dispatch[14]loop_start)も no-opt になるので、 OSR ループ JIT が二度と起動しない

以上でオフスタックのコードは片付くが、今スタックに乗っている JIT フレームは まだ古い本体を実行し続ける。その始末は set_bop_redefine の中ではなく、 呼び出し元の Executor::{add_method, add_method_with_original, alias_method_for_class} が担う。メソッド表を書き換えたmethod_added フックを呼ぶCodegen::check_bop_redefine(cfp) を通し、フラグが非 0 なら Codegen::chain_deopt_into が CFP 鎖を遡って各サスペンド中フレームを インタプリタフレームへ変換する(write-back の replay + return address を 共有 VM continuation stub に書き換え。doc/chain_deopt.md §10 を参照)。 戻ってきた時点で VM に落ちる。かつてはコードそのものに jmp deopt を 書き込む immediate_eviction がこれを担っていたが、chain deopt が完全に 置き換えたので自己書き換えコードは無くなった。 method_added は任意の Ruby を走らせるので、その前に片付けておく必要がある。

この配置には 2 つ性質がある。

  • レベルトリガであってエッジトリガではない。 check_bop_redefine は 「今回の定義が basic op を潰したか」ではなく「フラグが立っているか」を見る。 フラグは一度立つとクリアされないので、以後プロセス内のあらゆるメソッド定義が 毎回 CFP 鎖の全走査とフレーム変換を行う。実測では他の要因 (JIT が止まることで再コンパイル churn も消える)に埋もれて有意差は出なかったが、 構造としては無駄が残り続ける。
  • 経路が 3 つの funnel に限られる。 remove_method はここを通らない (§1.4 の但し書き)。

JIT コード内でのランタイム検査は AsmInst::CheckBOP(フラグのロード + 分岐) だが、発行箇所は MethodDef / SingletonMethodDef の直後のみ (jitgen/compile.rs)。演算そのものにはガードが無い — これは意図的で、 定数畳み込み(100 * 10010000)とレジスタ常駐を成立させるための選択 (jitgen/compile/binary_op.rs のコメント参照)。

1.4 網羅の穴

インライン展開しているのに basic op として登録されていない演算子が多数ある。 48 ケースを CRuby 4.0.2 と差分テストした結果、22 件が食い違った

VM ティアで再定義が完全に無視されるもの:

Integer   %  **  <<  >>  &  |  ^  ==  <  <=  >  >=  -@
Float     /  ==  <  >
String    ==
Array     []        Hash  []        Symbol  ==

JIT ティアは挙動が分かれる:

  • 正しいInteger#%Array#[]Hash#[]Symbol#==BinOpK::{Shl, Shr, Exp, Rem} は「常にメソッド呼び出しにコンパイル」する 分岐に入り、インラインキャッシュ + class-version ガードを経由するため。
  • 誤り& | ^、比較全般、-@binop_integer / 比較の無条件インライン 展開で、ガードが無い。

send(:+) 経由は常に正しい(メソッド表を引くため)。構文としての演算子だけが 壊れる。

上記 22 件のうち 1 件(Array#size)は種類が違う。monoruby は Array#size を Ruby で実装しているため、再定義が Kernel#p の内部を壊して ArgumentError になる。BOP フックの欠落ではなく「ビルトインが Ruby 実装で あることの露出」で、CRuby(C 実装)には無い問題。切り分けて扱うべき。

remove_method は検知経路そのものが無い。 検知は ClassInfoTable::insert_method(「上書きされた古いエントリが basic op か」)に だけ置かれているが、ClassInfoTable::remove_methodmethods.remove() を 直接呼び、is_basic_op を一切見ない。結果:

操作monorubyCRuby
Integer.remove_method(:+)1 + 23NoMethodError
Integer.undef_method(:+)1 + 2NoMethodErrorNoMethodError
Integer.alias_method(:+, :-)1 + 2-1-1

undef_method が通るのは add_empty_method 経由で insert_method を踏むため。 CRuby は追加・削除・prepend のいずれでも rb_vm_check_redefinition_opt_method を引くので取りこぼさない。検知点は 「メソッド表を変更する全経路」に置く必要がある。

1.5 コスト(実測)

fib(30)Float#+ を再定義。ワークロードは Float を一切使わない。

条件時間
JIT・再定義なし0.022 s
--no-jit(純 VM)・再定義なし0.069 s
JIT・Float#+ 再定義後0.51 s
JIT・再定義後に定義したメソッド0.52 s

無関係なクラスの再定義 1 回で JIT ありの 24 倍、純 VM の 7.5 倍遅くなる。 _no_opt ハンドラは fixnum インラインパスを捨てて毎回フルディスパッチするため、 「JIT を止める」より悪い。再定義後に定義したコードも救われない (remove_vm_bop_optimization の効果はプロセス全体・恒久のため)。


2. CRuby の戦略

  • ruby_vm_redefined_flag[BOP_xxx] — BOP ごとに 1 ワード、その中のビットが クラスINTEGER_REDEFINED_OP_FLAGFLOAT_…STRING_…ARRAY_…HASH_…SYMBOL_… …)。粒度は (演算子, クラス) の組
  • vm_opt_method_defs(クラス, メソッド名) → BOP の対応表。 rb_vm_check_redefinition_opt_method がメソッドの追加・削除のたびに引き、 一致したときだけ該当ビットを立てる。
  • インタプリタopt_plus 等の中で、オペランドの型チェックのBASIC_OP_UNREDEFINED_P(BOP_PLUS, INTEGER_REDEFINED_OP_FLAG) (ロード + AND + 分岐)を実行する。実行のたびに検査するが、対象は 1 ワード なので分岐予測がほぼ完全に効く。
  • YJIT / ZJITassume_bop_not_redefined() で「その (class, bop) は未再定義」 という invariant にコードブロックを登録する。再定義時はその invariant に 依存するブロックだけを無効化する。
  • 通知 — Ruby 3.4+ は -W:performanceRedefining 'Integer#+' disables interpreter and JIT optimizations を出す。

つまり CRuby はインタプリタでは実行時チェック、JIT では invariant + 局所無効化 という二本立てで、どちらも粒度は (演算子, クラス) である。


3. 比較(fib(29) × 3 回、実測)

baselineFloat#+ 再定義後String#+ 再定義後
CRuby0.062 0.061 0.0590.059 0.059 0.0610.058 0.059 0.059
monoruby0.014 0.013 0.0130.310 0.321 0.3280.012 0.012 0.012

CRuby には測定できるほどの影響が無い。monoruby の String#+ が無影響なのは、 そもそもフックが無い(未登録)からであって、良い意味ではない。

CRubymonoruby
粒度(演算子, クラス)グローバル 1 ビット
検査方式fast path 内の実行時テスト(VM)/ invariant(JIT)事前の一括無効化
影響範囲その組を使うコードのみプロセス全体
可逆性不要(他が影響を受けない)不可逆
網羅約 30 op × 十数クラス10 エントリ
未登録 opそもそも fast path を持たないfast path はあるがフックが無い → 誤答
通知-W:performance無言

4. 評価 — 何が正しく、何が間違っているか

戦略は正しい。 monoruby の JIT が採る「ガード無しでインライン展開し、 再定義時に無効化する」は YJIT と同じ方向であり、CRuby インタプリタ式の 「毎回フラグを読む」を JIT に持ち込むと、いま効いている定数畳み込みと レジスタ常駐が成立しなくなる。ここは維持すべきである。

間違っているのは粒度と網羅である。

  1. 粒度 — 「どれか 1 つでも再定義されたか」しか持たないので、反応は プロセス全体を落とすしかない。bop_redefined_flags が既に u32 である にもかかわらず、0 か全ビットかの真偽値としてしか使われていない。
  2. 網羅 — インライン展開している (クラス, 演算子) の大半が未登録で、 再定義が無言で無視される。速い誤答は遅い正答より悪い。
  3. 恒久性jit_invalidated も VM ディスパッチ表の差し替えも一方向で、 回復手段が無い。粒度 1 の設計から必然的にこうなっている。

5. 方針

Step 1 — 網羅を閉じる(正しさ、最優先)— 実装済み

globals/store/basic_op.rs62 組の (クラス, メソッド) の静的表BASIC_OP_DEFS)を置き、検知をそこに移した。CRuby の vm_opt_method_defs と同型である。

設計上の要点は 判定キーを「上書きされたエントリのフラグ」から「(クラス, メソッド) の組」に変えたこと。§1.4 の内訳を見ると、Integer#! / #+@ / #~NilClass#== などはそのクラスにエントリを持たないObject / BasicObject からの継承)。class Integer; def !; … は上書きではなく 挿入なので、いくらエントリにフラグを立てても旧方式では原理的に検知できない。 CRuby が静的表を引くのも同じ理由による。

あわせて:

  • remove_method にも検知を入れた(§1.4 の但し書き。undef_methodadd_empty_method 経由で insert_method を踏むので既に通っていた)。
  • ブートストラップ用ラッチ armed を追加。ビルトイン自身の定義がこの表そのもの なので、startup.rb とgem のロードが終わるまで報告しない (Store::arm_basic_opsExecutorstartup_flag と同じ地点)。
  • ディスパッチ表を持たない Rust 側の fast path(runtime::{get_index, set_index}Array#[] / Hash#[] / Array#[]=)は _no_opt 版に 差し替えられないので、BasicOpTable::redefined を直接読ませた。

結果(実測):

変更前変更後
演算子再定義スイープ(257 ケース)51 件が CRuby と食い違い0 件
remove_method(:+)1 + 23NoMethodError
core 全体スペック(単一プロセス)347 F / 219 E345 F / 218 E
cargo test59/5959/59
--features emit-asm(BOP 再定義なし)バイト単位で同一

Step 1 が露出させたもの — Ruby 実装ビルトインの脆さ — 解決済み(#1135)

: 本節は問題の記述である。最後に解決の記録を足してある。

再定義が実際に効くようになった結果、monoruby が Ruby で書いている ビルトイン(builtins/*.rb、約 9,000 行)がその再定義を踏むようになった。

class Integer; def <(o); :OV; end; end
[1, 2, 3].map { |x| x * 2 }
#   CRuby    => [2, 4, 6]        (Array#map は C)
#   monoruby => NoMethodError    (array.rb の `while i < size` が壊れる)

Integer.remove_method(:<) はさらに露骨で、Comparable#<comparable.rb)が res < 0 で自分自身に再帰し StackOverflow になる。CRuby の Comparable#< は C なのでディスパッチしない。

これは Step 1 が作った欠陥ではなく、Step 1 が可視化した既存の構造的弱点で ある。変更前は再定義そのものが無視されていたので、ユーザのコードでも ビルトインでも一様に「元の演算子」が使われ、辻褄だけは合っていた。

トレードオフを正直に言えば、Step 1 は「1 < 2 が誤答」を「1 < 2 は正答だが Array#map が壊れる」に置き換えた。前者はサイレント、後者はラウド。 cargo test と ruby/spec には影響が無い(どちらも組み込みクラスの演算子を 再定義しない)ため実害は測定されていないが、基本演算を monkey patch する 実プログラムは動かない

塞ぐには、builtins/*.rb の演算子をビルトインに束縛する(= CRuby が C で 書くことで無料で得ている性質を明示的に作る)必要がある。粒度とは独立の課題 なので Step 2 の前提ではないが、Step 1 の帰結として記録しておく。

解決(#1135)

その束縛を入れた。builtins/*.rb 由来のフレームにはもともと 1 ビットが 立っている(Meta::set_internal_builtin$~/$_ を持たせないためと、 refinements を届かせないため —— 後者は #1066 で、まったく同じ理由で 同じ判断をしている: 「Array#map 自身の i += 1 はライブラリのコードで、 CRuby では C だから refinement には見えない」)。再定義もそこに届かせない。

読む場所は 3 つ:

場所役割
Executor::dispatch_redefined_opVM ヘルパ(*_values 系)のガード。内部ビルトインのフレームからなら _raw へ落とす
JitContext::resolve_basic_opJIT の fast path 解決。生のメソッド表ではなく arm 時のスナップショットStore::basic_op_armed_func)を引く
JitContext::basic_op_assumable上で解決したものをガード無しで inline してよい、という licence

VM 側だけでは直らない。JIT は jit_check_methodInteger#+生の メソッド表から引き、再定義後はユーザのメソッドに解決して inline generator を 失う —— つまり basic_op_assumable に到達すらしない。スナップショット解決が その半分である。

差分スイープ(8 種の再定義 × 28 プローブ = 224 セル、CRuby 4.0.2 と比較): CRuby と異なるセルが 30 → 1

残る 1 セルは Integer#< を再定義したときの 5.times: CRuby は [] を返す。 CRuby の Integer#times も Ruby で書かれているためである。推定ではなく CRuby 自身に吐かせた:

$ ruby -e 'p Integer.instance_method(:times).source_location'
["<internal:numeric>", 255]        # upto / Array#each は nil(= C)

その iseq の逆アセンブル(ループ本体):

0020 getlocal_WC_0   i
0022 invokeblock     <argc:1>                        # yield i
0025 getlocal_WC_0   i
0027 opt_succ        <calldata!mid:succ, argc:0>     # i = i.succ
0029 setlocal_WC_0   i
0031 getlocal_WC_0   i
0034 opt_lt          <calldata!mid:<, argc:1>        # while i < self
0036 branchif        20

+ を呼ぶ箇所が無いので Integer#+ の再定義は届かない。succ< は実際に 呼んでいるので届く —— opt_succ / opt_lt は BOP 高速命令だが、再定義されれば BASIC_OP_UNREDEFINED_P が落ちて本物のディスパッチに戻る。monoruby が Step 1 以降 踏んでいたのとまったく同じ構造である。裏取りとして succself + 2 に再定義 すると CRuby の times[0, 2, 4] を返す。

monoruby は + / < / succ のいずれに対しても免疫になった。CRuby の同種の 瑕疵をわざわざ再現しない、という判断である。

将来の選択肢 — intrinsic による粒度の精密化(見送り)

フレーム単位の束縛は「このファイルの演算子は全部ビルトイン束縛」としか言えず、 ライブラリ自身の帳簿i += 1)と呼び出し側のための演算sort<=>)を区別できない。区別は今のところ手作業で保っている (EqSearch::newop/sort.rs は意図的に束縛の外)。

原理的に正しいのは issue #1135 が第一候補として挙げた方向 —— __add / __lt のような再定義されない内部呼び出しを導入し、 1 箇所ずつ明示する。実装するなら「本物のメソッド」ではなく bytecodegen が名前を認識して通常の BinOp 命令に落とす形にすること。 本物のメソッドにすると VM 階層の演算子高速路を失う(i += 1i.succ で 増分が実測 1.8 倍)。名前を「再定義チェックを外す印」として使い命令自体は 同じにすれば、性能は不変になる。

規模(実測):

CRuby のコアメソッドのうち Ruby 実装74 / 1061(7%)
— うちループを回すものInteger#times のみ
builtins/*.rb の演算子使用約 1,058 箇所 / 11,217 行
— うちループ帳簿(+= 1 / -= 1 / while … <約 150(機械的)
— 残り(== 199, < 168, <=> 43 …)約 900(1 件ずつ判断が要る)

CRuby 側が C なのは 93% なので「CRuby が C のものだけ intrinsic 化」は事実上 ほぼ全部が対象になる。そして残り約 900 箇所の多くは呼び出し側の値を比較して おり、intrinsic 化してはいけない —— 判断を誤ると今回のバグと逆向きに、 同じくサイレントに壊れる。

したがって一度に移行はしない。フレーム単位の束縛を既定の安全側とし、粒度が 粗すぎて実害が出た箇所から漸進的に intrinsic 化する方針とする(2026-09、 #1362 のレビューで決定)。なお CRuby との完全一致だけが目的なら、intrinsic より安い道がある —— Ruby 実装側の 74 件を束縛から除外する方向で、Integer#times なら「印から除外」+「増分を i.succ に変更」の 2 箇所で + / < / succ の 3 セルとも一致する。こちらも採らない(CRuby の瑕疵の再現になるため)。

Step 2a — VM を (クラス) 粒度にする — 実装済み

鍵になった観測: VM の asm fast path はすべて fixnum 限定である。 vm_binops_opt / vm_cmp_opt! / vm_neg … はどれも guard_rdi_rsi_fixnum / guard_rdi_fixnum で始まり、fixnum でないオペランドは 必ず Rust ヘルパ(add_valuescmp_lt_valueseq_values_visnot_valueget_index …)に落ちる。

したがって:

  • Integer の再定義 → asm が無効になる → ディスパッチ表の差し替えが要る。
  • それ以外のクラス(Float / String / Symbol / nil / true / false / Complex / Array / Hash)→ asm は書かれたとおり正しいまま。無効になるのは Rust ヘルパのそのクラス用アームだけなので、ディスパッチ表に触る必要がない

実装は 2 点だけ。

  1. BasicOpTable が「何が再定義されたか」を持つ(redefined_setinteger_redefined)。Store::set_bop_redefineInteger が初めて 再定義されたときだけ remove_vm_bop_optimization() を呼ぶ。
  2. ネイティブに答える Rust ヘルパが、そのアームに入る前に basic_op_redefined_for(受け手のクラス, 演算子名) を確認する。 グローバル bool で門番しているので、再定義しないプログラムの追加コストは bool 1 個(しかも呼び出し自体が既に C-ABI)。

対象ヘルパ: binop_values!(add/sub/mul)、手書きの div_values / rem_values / pow_values / shl_values / shr_valuescmp_values! (lt/le/gt/ge)、eq_values_viscmp_teq_values_implnot_value / neg_value / pos_value / bitnot_valueget_index / set_index。 (!=custom_neq が既にメソッド表を引くので変更不要。)

結果(fib(29) × 3、実測):

条件Step 1 までStep 2a
baseline(JIT)0.0090.009
Float#+ 再定義後0.310.030
String#== 再定義後0.310.031
Array#[] 再定義後0.300.030
Symbol#== 再定義後0.300.030
Complex#* 再定義後0.310.030
(参考)--no-jit0.0300.030
(参考)CRuby0.0510.051

再定義後の 0.030 は --no-jit の 0.030 と完全に一致する。 つまり VM 側の 劣化は完全に消え、残る 3.3 倍(0.009 → 0.030)はまるごと JIT の グローバル無効化である。これが Step 2b の対象。

正しさは 257 ケースのスイープが 0 差分を維持、cargo test 59/59、core 全体 スペックは 345F/218E → 344F/218E--features emit-asm は Step 1 前の master とバイト単位で同一のまま。

Step 2c — VM を (演算子) 粒度にする — 実装済み

Step 2a の後も、Integer の再定義だけは依然としてディスパッチ表の一括差し替えを 起こしていた。Integer#~(fib が一切使わない)を再定義するだけで 0.013 → 0.33 s(25 倍、--no-jit の 7.6 倍悪い)。粒度がクラス単位までしか 無かったためである。

解法はフラグをアセンブリから読ませること — CRuby インタプリタの BASIC_OP_UNREDEFINED_P と同じ配置。VmBop ごとに 1 ワードを JIT データ領域に 置き(Codegen::bop_flags)、各 VM ハンドラが自分の fixnum ガードの直後で 自分のワードを読む。非 0 なら inline 計算を諦めて generic パス(Rust ヘルパ)へ 抜ける。

ガードを持つのは asm の inline 計算を持つ 11 個だけ: add sub eq ne lt le gt ge === +@ -@~! は元々 inline 計算が無く、 残りの binop(* / % ** << >> & | ^)は asm 高速路自体を持たず最初から Rust ヘルパ直行なので、Step 2a のペアチェックで足りる。

副産物として _no_opt 系が丸ごと不要になった。 差し替え先が要らなくなった ので、両アーキの VM _no_opt ハンドラ群と、*_values_no_opt / *_value_no_opt の Rust ヘルパ群(マクロごと)を削除した。 remove_vm_bop_optimizationdispatch[14]loop_start)だけを扱う disable_vm_loop_jit に縮小した — これは basic op ではなく JIT 無効化の話で、 Step 2b の対象。

結果(fib(29) × 3、実測):

条件Step 1Step 2aStep 2c
baseline(JIT)0.0090.0090.009
Float#+ 再定義後0.310.0300.030
Integer#~ 再定義後0.330.330.032
`Integer#` 再定義後0.330.33
Integer#* 再定義後0.330.330.031
(参考)--no-jit0.0300.0300.030

すべての再定義が --no-jit 相当に収まった。 残る 3.3 倍(0.009 → 0.030)は まるごと JIT のグローバル無効化= Step 2b。

ガードの実費(正直な数字):

モードガード無しガード有り
通常(JIT)fib 0.0020 / loop 0.0131fib 0.0021 / loop 0.0137
--no-jit、算術飽和ループ(15 回交互、中央値)0.2300.246(+7%

通常モードでは min / median が一致し測定できない。VM 専用の合成ベンチでのみ 約 7% で、その大半は asm ガードではなく Rust ヘルパ側の basic_op_redefined() 参照(& ^ | を多用するため)と見られる。VM は通常 ウォームアップ経路なので、10 倍の崖を消す対価として受け入れた。

フラグは 1 ワード・1 演算子 1 ビット。 当初は VmBop ごとに data_i32 を 1 語ずつ(11 語 44 バイト)確保していたが、これは CRuby の ruby_vm_redefined_flag がビットマップである理由をなぞり損ねていた。1 語 (data_i64)にまとめ、VmBop は語の添字ではなくビット番号を意味するようにした。 x86 は cmpl [rip+f], 0testq [rip+f], (1<<bit)、aarch64 は ldr w9/cbnzldr x9/tbnz #bit で、どちらも命令数は不変。ガードが全部 同じキャッシュラインに乗り、bop_flags: Vec<DestLabel>VmBop::COUNT が 消え、残る 53 ビットが Step 2b の (op, class) 用に空く。

Step 2a の取りこぼし — ガードを「実装」に置いてしまっていた。 add_values などの共有ヘルパは VM/JIT の generic パス(メソッド探索を 飛ばした呼び出し元)だけでなく、ビルトイン Integer#+ の本体そのものでも ある。ガードをその共有ヘルパの先頭に置いたため、

class Integer; alias __orig :+; def +(o); __orig(o); end; end
1 + 2   # => StackOverflow

__orig(= 元のビルトイン)に入った時点でガードが再び発火し、名前 + で 再ディスパッチして利用者の + に戻る、という無限再帰になっていた。CRuby が BASIC_OP_UNREDEFINED_P を命令側(opt_plus)に置き rb_int_plus には 置かないのは、まさにこの理由である。

修正は同じ形に揃えること — 実装を *_values_raw として無防備なまま残し、 ガードは「探索を飛ばした呼び出し元のための入口」*_values に分離した。 ビルトインの本体(numeric.rsbinop!/unop!integer.rscmpop!% ** << >>float.rs の比較と ==)は _raw を呼ぶ。 Array#sumNumeric#angle のように探索せずに演算子を使う内部呼び出しは ガード付きのままでよい(CRuby も同じ位置で BOP を見る)。 検出には alias スイープ(253 ケース)を使い、39 → 10 = master と同数・同一集合まで 戻した。回帰テストは aliasing_an_operator_before_redefining_it_does_not_recurse

この手法の再利用先 — TracePoint。 「JIT データ領域に 1 ワード置き、生成 コードの決まった地点でそれを読んで分岐する」という形は basic op 固有ではなく、 実行中に切り替わりうるグローバルなスイッチ全般に効く。次にこれを必要とする のは TracePoint(および set_trace_func)で、line / call / return / b_call などのイベントが有効かどうかを VM ハンドラが自前で判定できるように なる。

CRuby はここでは別の道を採っていて、TracePoint 有効化時に iseq の命令列を trace_* 変種へ書き換える(vm_trace_setup)。無効時のコストが厳密に 0 に なる代わりに、命令ごとに並行するハンドラ集合を維持する必要がある — つまり Step 2c でちょうど削除した _no_opt 系と同じ構造の重複を、命令表全体に対して 背負うことになる。フラグ読みは load + 分岐 1 個を払うが、上表の通り通常 モードでは測定にかからず、ハンドラは 1 系統のままで済む。monoruby には こちらが合う。

検証: 257 ケースのスイープ 0 差分(途中 Integer#& | ^ が 3 件後退した — これらは int_binop_values! という別マクロ由来でガードから漏れており、 スイープが検出した)、cargo test 59/59、JIT の出力コードは バイト単位で同一(ページ内オフセットのみ 160 バイト移動)、core 全体スペックは 344–345F / 218–219E で同一バイナリでも実行ごとにこの幅で振れるため回帰なし。

Step 2b — JIT を invariant 単位にする — 実装済み

Step 2c まで終えても、baseline 0.009 に対してあらゆる再定義が 0.030 に 落ちていた。原因は 1 か所 — set_bop_redefine がプロセス内の 全 iseq の JIT コードを一方向ラッチ (jit_invalidated) で恒久的に捨てていたこと。 Array#[] を再定義しただけで fib の JIT が消える。

依存を記録して、該当分だけ捨てる。 JIT がガード無しに basic op を仮定して いるのは、整数・浮動小数の算術/比較/定数畳み込みと単項演算だけである (state/binop.rsbinop_integer / binop_float / gen_cmp_* / unop_integer_* / unop_float — いずれも per-method の inline 生成器から 呼ばれ、fire_binary_inline / fire_unary_inline が license を確認・記録 する)。それ以外 — メソッド呼び出しに落ちるものすべて — は既に class-version ガードで 守られている(再定義は必ず class version を進める)。そこで:

  1. JitContext::assume_basic_op(class, op) を inline 経路の入口に置き、
    • 再定義済みなら false を返してその演算だけ通常のメソッド呼び出しへ 降格する。だから再定義後に再コンパイルしても 健全で、しかも他の演算子の inline は保たれる
    • まだビルトインなら true を返しつつ (class, op) を記録する。
    • (現在は binop/cmp のディスパッチが inline 生成器経由 (fire_binary_inline)になり、純検査 basic_op_assumable + 生成器成功時のみの record_bop_dep に分離されている。融合 compare-and-branch もこの経路で license を確認・記録する。)
  2. 記録は jit_compileISeqInfo::bop_deps(self class をまたいで union)。
  3. set_bop_redefineevict_jit_code_for_bop(class, name)依存している iseq だけを捨てる。x86 の entry 巻き戻しも同じ集合に限定。
  4. jit_invalidated の意味を分離した。従来はフロントエンドの bail(恒久的に コンパイル不能)と「コードを捨てた」を同じフラグで表していた。前者は invalidate_jit_code、後者は回復可能な evict_jit_code とし、捨てた 本体は次のウォームアップで再コンパイルされる。
  5. check_bop_redefineレベルトリガを解消(§1.3)。bop_eviction_pendingset_bop_redefine が立て、直後の check_bop_redefine が消費する。 従来は sticky フラグを読んでいたため、一度どれかを再定義したプログラムでは 以後すべての def が制御フレーム鎖を全走査して、既にパッチ済みの return address を patch し直していた。
  6. AsmInst::CheckBOP世代比較にした。JIT データ語を「何か再定義された」 フラグからカウンタに変え、コンパイル時の値を焼き込んで比較する (class-version ガードと同じ形)。従来は 0 比較の sticky だったので、 一度でも再定義が起きるとコンパイル済みコード中の def 地点が永久に deopt し続けた。

結果(fib(29) × 3、実測):

条件Step 2cStep 2b
baseline(JIT)0.0090.009
Float#+ 再定義後0.0300.009
Integer#~ 再定義後0.0320.009
Integer#| 再定義後0.0310.009
Integer#* 再定義後0.0300.009
Array#[] 再定義後0.0310.009
(参考)--no-jit0.0300.031

崖が消えた。 fib が使わない演算子の再定義は完全に無料になった。

検証: 再定義スイープ 257 件 0 差分、alias スイープ 253 件 10(master と同数・ 同一集合)、cargo test 全 59 スイート 0 失敗、aarch64 クロスチェック通過。 回帰テストは redefining_an_operator_a_compiled_method_inlined(ウォーム済み JIT コードが使っている演算子を再定義しても正しい)と redefining_an_unrelated_operator_leaves_compiled_code_alone

Step 2b-2 — OSR ループ本体も iseq 単位にする — 実装済み

Step 2b の時点では、stale な本体が 1 つでも出ると dispatch[14]loop_start)を no-opt に落としてプロセス全体のループ JIT を止めていた。 コンパイル済みループ本体の codeptr はバイトコード内 LoopStart[pc+8] に埋まっていて、他から辿る口が無かったためである。

そのオペランドを直接ゼロにすればよい。 evict_jit_code が、捨てる iseq の バイトコードを走査して LoopStart(opcode 14)の [pc+8] をクリアする (clear_loop_jit_entries)。サイトは「まだコンパイルしていない」状態に戻り、 次の周回で再コンパイルされる — 今度は再定義された演算子を inline せずに。

オペランドは三値である点に注意: 0 = 未コンパイル、1 = コンパイルが bail した恒久センチネル、それ以外が実エントリ。クリアするのは 3 番目だけで、 センチネルを戻すと bail するループが閾値を跨ぐたびに(失敗する高価な) コンパイルを再試行してしまう。[pc+0] のヒットカウンタは閾値のまま残すので、 再ウォームアップを待たず次の周回で再コンパイルされる。

これで disable_vm_loop_jit / vm_loop_start_no_opt は不要になり削除した。

結果(無関係なループ、3M 周 × 3):

loop_mulloop_add を両方温めてから、loop_mul だけが使う Integer#* を(意味を変えずに)再定義し、loop_add を測る:

loop_add(無関係)
変更前0.027
変更後0.006

4.5 倍。 無関係なループが巻き添えで OSR JIT を失うことは無くなった。

Step 2b(当初の計画・記録として保存)

jit_invalidated のグローバル一方向ラッチをやめ、「この iseq がどの (op, class) invariant に依存したか」を記録して該当分だけ無効化する。 InlineCacheEntry と class-version ラベルという前例がある。あわせて dispatch[14]loop_start)の no-opt 化とオンスタック始末 (現 chain_deopt)のレベルトリガ(§1.3)もここで直す。

Step 2 — 粒度を (演算子, クラス) へ(性能)

bop_redefined_flags: u32BOP ごとのワード × クラスビットに変える (CRuby と同型)。

  • VM: _no_opt への全面差し替えをやめ、各 fast path で該当ビットだけを テストする。fixnum パスに分岐 1 個を足すコストは、現状の 7.5 倍劣化に比べれば 無視できる。
  • JIT: jit_invalidated のグローバル一方向ラッチをやめ、「この iseq が どの (op, class) invariant に依存したか」を記録して該当分だけ無効化する。 InlineCacheEntry と class-version ラベルという前例がある。
  • オンスタック始末(現 chain_deopt)は残す — オンスタックのフレームを 片付ける手段は粒度に関係なく必要である。ただし §1.3 の 2 性質を直す: 「フラグが非 0 か」の レベルトリガをやめてマスクのビットが 0→1 に遷移したときだけ走らせ (エッジトリガ)、走査対象もその (op, class) に依存したフレームに絞る。
  • 効果: 「無関係な Float#+ の再定義で fib が 24 倍遅くなる」が消える。

Step 3 — refinements の基本演算(#1066)— 実装済み

Step 1・2 を終えた時点で、残作業はペアを記録することだけだった。実験で 確認できる: 意味を変えない再定義(alias __p +; def +(o) = __p(o))で フラグだけを立てると、refine Integer { def +(o) = 42 } が VM・JIT とも CRuby と同じ 42 を返した。下流はすべて既に動いていた — Executor::find_method は refinement 対応、JIT は assume_basic_op が false を返して通常呼び出しへ降格し jit_check_methodJitContext::refinements の下で解決する。

検知点は refine 側ではなく insert_method / remove_method に置いた。 refinement のメソッドは refinement モジュール側の ClassId に入るので、 refined_class() を引いて (refined_class, name) で照会する。この位置なら def / define_method / alias / import_methods を 1 箇所でカバーできる (Step 1 の「メソッド表を変更する全経路に検知点を置く」と同じ理由)。

必要だった前提修正 — ライブラリ境界。 Executor::frame_refinements は monoruby が Ruby で書いたコアライブラリの フレームを透過して呼び出し側スコープまで歩く。これは &objto_proc や補間の to_s のためで、CRuby がそれらを呼び出し側の iseq で 行うのに対し monoruby は callee 側で行うため、透過しないと一致しない。

ところが演算子はその種の変換ではないArray#map 自身の i += 1 は ライブラリのコードで、CRuby では C — どの refinement からも見えない。 呼び出し側スコープで解決した結果、refine された Integer#+ が 42 を返して map が 1 周で終了していた([1,2,3].map { |x| x*2 }[2, nil, nil](1..3).sum → 42)。

そこで find_method が解決対象の名前で分岐するようにした: BASIC_OP_DEFS の演算子名ならライブラリ境界で歩みを止め (Executor::basic_op_refinements)、それ以外は従来どおり透過する。

Step 3b — JIT をスコープ単位にする — 実装済み

Step 3 直後はペアの記録がプロセス全体だったので、refine しただけで (using していないスコープでも)その演算子の inline を失っていた (fib(29) 0.009 → 0.034)。

「何が無効化したか」を分けて記録する。 BasicOpTableredefined_set(= fast path はもう無条件には健全でない、という和集合)を 残したまま、内訳として globally_redefined_setrefined_set を持つ。 assume_basic_op は 2 段で問う:

  1. グローバル再定義なら、どのスコープも逃れられないので inline しない。
  2. refinement 由来なら、コンパイル中のスコープの RefinementSetId が 実際にそのペアを解決し直すかだけを問う(basic_op_refined_in_scope)。 refine していないスコープ — refine しないプログラムの全スコープを含む — は inline を維持する。

判定は「set のエントリのうち当該クラスを refine するものを辿り、その refinement モジュール(と ancestors)が当該名を持つか」で行う。当初は 「set 有り/無しで解決して比較する」正確版を書いたが、無し側が check_method_for_class → メソッドキャッシュ → Globals::class_version() と 辿って JIT コンパイル中に CODEGEN を borrow し panic した(コンパイル時は 既に borrow_mut 中)。エントリ走査ならクラス表しか触らない。判定は 意図的に広めに倒してある — 迷ったら実呼び出しにするのは常に健全で、逆は refinement が置き換えたはずの算術をそのまま出してしまう。

条件Step 3Step 3b
baseline0.0090.009
refine(Integer) { def + } 後(using 外)0.0340.009
refine(Integer) { def ~ }0.0340.009
(参考)グローバル再定義後0.0320.032

refine しただけのコストが消えた。 VM 側の asm ガードは呼び出し地点の 文脈を持たないグローバル語なので粗いまま(正しさは dispatch が担保する)。

Step 4 — mixin 経由の再定義(#1214, #1219)— 実装済み

Step 1 で「メソッド表を変更する全経路に検知点を置く」と決めたが、経路が 1 本抜けていた。モジュール経由である。

module M
  def +(o) = super(o) * 2
end
class Integer
  prepend M
end

insert_method が受け取る class_idM であって Integer ではない。 表が記録しているペアは (Integer, "+") なので contains(M, "+") は false、 何も立たず両ティアがビルトインを撃ち続けた。順序を入れ替えて prepend を先にしても同じ(定義そのものが走らない)。

検知点は 2 つ:

  • insert_methodcheck_mixed_in_basic_op(module_id, name) — モジュールに後からメソッドが入った場合
  • append_features / prepend_featurescheck_mixin_basic_ops(module) — すでにメソッドを持つモジュールが後から差し込まれた場合

罠 — 「モジュールがその名前を定義しているか」は問いとして間違っている。 最初の実装はこう書いた:「name が BOP 名で、かつ module_id が基本演算 クラス C の祖先にあるなら (C, name) を再定義済みにする」。これは ユーザコードの include Comparable ひとつでプロセス中の整数 fast path が 全滅するComparable< <= > >= == を定義しており、 Integer はブートストラップ以来それを include している。つまり 「祖先のモジュールがその名前を定義している」はどんなプログラムでも常に真 で、一方 Integer 自身の < はルックアップで勝ち続けるので実際には何も 変わっていない

正しい問いは「そのクラスは今もビルトインに解決するか」である。 arm_basic_ops が起動時に全ペアの解決先 FuncId をスナップショットし (BasicOpTable::armed_funcs)、mixin 検知はそれと現在の解決先を比較する。 Integer.prepend M(M が + を持つ)なら解決先が動くので立ち、 class Foo; include Comparable; end なら動かないので立たない。

実測(yjit-bench lee, x86-64)。 この誤検知は 1 回の includeInteger#< <= > >= == != === <=> !FloatStringSymbol の同種、 計 27 ペアを恒久的に落とした。leerequire "json" の過程で Comparable を include するため直撃した:

median
誤検知の前(037a720512 ms
誤検知の入った master(21ed6aa1728 ms
解決先比較を入れた後520 ms

CI の履歴では amd64 が 478 ms → 1550 ms(YJIT 比 1.48x → 0.44x)、 arm64 は 400 秒のタイムアウトに達した。

回帰テストは basic_op.rsa_mixin_that_displaces_nothing_keeps_the_basic_ops / a_prepend_that_displaces_the_builtin_retires_its_pair。前者は「何も 落ちない」ことを、後者は「落ちるべきものは落ちる」ことを表の状態に対して 直接主張する — 値の比較では、fast path が消えても答えは正しいままなので 検出できない。

Step 3 の当初計画(記録として保存)

Step 2 の (op, class) ビットマスクができれば、#1066 が要求する 「refinement セットごとの BOP ビットマスク」はその自然な拡張になる。

逆に、今の 1 ビット設計のままでは #1066 は実装できない。 refinement は 「このレキシカルスコープでだけ再定義」であって、グローバルビットとは意味が 異なるためである。したがって #1066 の前提として Step 1・2 を先に行うべきで、 順序を逆にすると設計をやり直すことになる。

検討した候補 — hash / eql? を BOP にする(見送り)

Hash / Set のルックアップは hasheql? と連鎖するので、これらを BOP に すれば速くなるのではないか、という案。測定した結果、利得はゼロだった。

理由は BOP 表の性質そのものにある。表は「この仮定を置いてよい」という許可検知であって、速度は「仮定を使う fast path」を書いて初めて出る。そして hash / eql? については その fast path が既にあり、無条件に効いているValue::ruby_hash / Value::eqlvalue.rs)が ObjTy で分岐して Rust 側で 直接計算する:

ネイティブ計算(ディスパッチ無し)
ruby_hashFixnum, Flonum, nil/true/false/Symbol, BigInt, Float, String, Array, Hash, Range
eql同一 id, 両方 immediate, BigInt, Float, Complex, String, Array, Range, Hash

基本クラスのキーには消せるディスパッチが最初から無い。登録しても得るものは無く、 グローバル崖のトリップワイヤが増えるだけである。正しさの穴も無い(String#hash を再定義しても Hash のバケッティング・Array#hash とも CRuby と一致する。CRuby も基本クラスのキーは rb_str_hash 等で直接計算する)。

実測(各 300k 回ルックアップ):

monorubyCRuby
Hash[String]0.014 s0.028 s
Hash[Symbol]0.011 s0.016 s
Hash[Integer]0.016 s0.014 s
Hash[ユーザ定義 hash/eql?]0.030 s0.100 s
Set#include?0.016 s0.017 s

ただし派生案は有効。 hash / eql? を定義していない素のオブジェクト (identity ハッシュ)は ruby_hash_ アームに落ち、ルックアップのたびに Object#hash へディスパッチしている。5 項目中ここだけ monoruby が負ける:

monorubyCRuby
Hash[素のオブジェクト]0.031 s0.023 s
Set#include?[素のオブジェクト]0.031 s0.024 s
obj.hash 単体0.008 s0.016 s

obj.hash 単体では monoruby が 2 倍速いので、差はハッシュ計算ではなく ルックアップ 1 回ごとのディスパッチ往復にある。(Object, hash) を表に入れ、 _ アームで「このクラスの hash はまだ Object#hash か」を確かめてから identity ハッシュを直接計算すれば省ける(eql 側は既に id() 比較で短絡済み)。

派生案 — 実装済み。 Value::ruby_hash_ アーム(ディスパッチ経路)の 手前に「このレシーバの hash はまだビルトインの Kernel#hash か」を問う アームを足し、そうなら Kernel#hash と同じ digest を直接計算する。

判定は BOP 表ではなく FuncId の比較にした。表は静的な (クラス, 名前) なので (Object, hash) を入れても Object#hash 自体の再定義しか捕まえられず、 ユーザ定義クラスが自前の hash を持つ場合を取りこぼす。実際に必要なのは 「このレシーバのクラスにおける hash の解決結果がビルトインか」で、これは Store::check_method(class-version キーのキャッシュ付き)1 回で答えられる。 表に入れると set_bop_redefine が発火して JIT まで巻き添えにするので、 むしろ有害だった。CRuby も rb_any_hash で同じ特別扱いをしている。

実測(同一ベンチの A/B、300k 回ルックアップ):

変更前変更後CRuby
Hash[素のオブジェクト]0.0500.0390.024
Set#include?[素のオブジェクト]0.0490.0380.026

22% 改善したが CRuby にはまだ届かない。 消えたのはディスパッチ往復だけで、 残る差はハッシュ表の実装そのものにあると見られる(obj.hash 単体では monoruby が 2 倍速いままなので、digest の計算コストではない)。ここから先は BOP とは別の課題。

正しさは CRuby と一致を確認 — 同一性による引き当て、hash/eql? を持つ クラスの値による引き当て、あとから hash を定義した場合の追随(fast path が外れる)、SetArray#hash の digest 一貫性。

併せて検討するもの

  • -W:performance 相当の警告 — 実装は小さく、デバッグ価値は高い。現状の monoruby は完全に無言で 24 倍遅くなる。
  • Ruby 実装ビルトインの露出(§1.4 但し書き)— BOP とは別問題だが、Step 1 で 誤答が減ると顕在化しやすくなるので、同時に把握しておく。

検討して採らなかった案

  • JIT にも実行時チェックを入れる(CRuby インタプリタ方式) — 定数畳み込みが できなくなる。invariant + 無効化を維持する。
  • jit_invalidated を解除可能にする(再コンパイル許可) — 複雑な割に、 Step 2 を行えば「そもそも無効化されない」ので不要になる。
  • basic op 登録をやめて全演算子をメソッド呼び出しにする — 正しさは得られるが、 monoruby の性能特性そのものを捨てることになる。

6. 受け入れ条件(Step 1・2 共通)

doc/refinements.md で用いた手法をそのまま使う。

  1. コード生成の同一性 — BOP 再定義を含まないワークロードで --features emit-asm の出力が変更前とバイト単位で一致すること。
  2. 差分テスト — §1.4 の 48 ケースが CRuby と一致すること(Array#size を除く)。
  3. 性能回帰なしfib / optcarrot が baseline を維持すること。
  4. Step 2 の効果Float#+ 再定義後の fib が baseline に戻ること (現状 0.51 s → 目標 0.022 s 近傍)。