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