-interpreter-instrument-with-probe

在 StableHLO 程序中插入 interpreter.probe 运算。

遍历 StableHLO 程序,并在每个合适的操作后插入一个探测插桩操作(请参阅下文,了解如何定义合适的操作)。插桩用于从 StableHLO 参考解释器中提取中间张量值,以便稍后与其他运行时进行比较。

除了常量运算和没有任何张量返回值的运算(即不会插桩产生元组或令牌或没有返回值的运算)外,所有运算都被视为适合插桩。无论嵌套级别如何,系统都会插桩适当的操作。也就是说,循环/分支区域内的操作也会被插桩。

在插桩操作执行时,其返回值将使用 NumPy 数据格式写入磁盘。如果启用了 useDebugInfo 传递选项,则系统会在可用时使用位置调试信息来唯一标识插桩张量值(即传递将从 NamedLoc(probe_id@<...>) 中提取 probe_id 并使用格式 probe_id.#)。否则,系统会按递增顺序引用插桩值:probe1probe2、... 如需详细了解数据的序列化方式,请参阅 interpreter.probe

选项

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