-interpreter-instrument-with-probe

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

StableHLO プログラムを一通り説明し、プローブ計測を挿入します オペレーションの後に適切なオペレーションを実行します（適切なオペレーションの失敗方法については、 定義します）。インストルメンテーションを使用して、 StableHLO 参照インタープリタのテンソル値（後で比較する場合に使用） 他のランタイムと連携できます

定数関数を除き、すべてのオペレーションが計測に適していると見なされる オペレーションはテンソル戻り値を持たないオペレーションです（つまり、 タプルまたはトークンが生成されない場合、戻り値は計測されません）。 適切なオペレーションは、API レベルに関係なく、 使用します。つまり、ループ/ブランチ領域内のオペレーションも します。

インストルメンテーション オペレーションでは、次のコマンドを使用して、戻り値がディスクに書き込まれます。 実行時に 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.