-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.