-interpreter-instrument-with-probe

StableHLO プログラムに interpreter.probe オペレーションを挿入します。

StableHLO プログラムを走査し、適切なオペレーションの後にプローブ計測オペレーションを挿入します（適切なオペレーションの定義については、後述をご覧ください）。計測は、後で他のランタイムと比較するために、StableHLO リファレンス インタープリタから中間テンソル値を抽出するために使用されます。

すべてのオペレーションは、定数オペレーション、テンソル戻り値のないオペレーション（タプルまたはトークンを生成するオペレーション、または戻り値のないオペレーション）を除き、計測に適していると見なされます。適切なオペレーションは、ネストのレベルに関係なく計測されます。つまり、ループ / ブランチ リージョン内のオペレーションも計測されます。

計測対象のオペレーションの戻り値は、実行時に NumPy データ形式を使用してディスクに書き込まれます。useDebugInfo パス オプションが有効になっている場合、位置情報のデバッグ情報が利用可能な場合は、計測対象テンソル値を一意に識別するために使用されます（つまり、パスは NamedLoc(probe_id@<...>) から probe_id を抽出し、probe_id.# 形式を使用します）。それ以外の場合、計測対象の値は probe1、probe2、... の順に参照されます。データのシリアル化方法については、interpreter.probe をご覧ください。

オプション

-useDebugInfo : Whether or not to use location debug data as `probe_id` values.