-interpreter-instrument-with-probe

Inserisce operazioni interpreter.probe in un programma StableHLO.

Esamina un programma StableHLO e inserisce un'operazione di instrumentation probing dopo ogni operazione idonea (vedi di seguito come viene definita un'operazione idonea). La misurazione viene utilizzata per estrarre valori intermedi del tensore dall'interprete di riferimento StableHLO per il confronto successivo con altri runtime.

Tutte le operazioni sono considerate idonee per la misurazione, ad eccezione delle operazioni costanti, delle operazioni che non hanno valori restituiti di tensore (ad es. un'operazione che produce una tupla o un token o nessun valore restituito non verrà misurata). Le operazioni idonee verranno messe in primo piano indipendentemente dal loro livello di nidificazione. Ciò significa che verranno instrumentate anche le operazioni all'interno delle regioni di loop/branch.

I valori restituiti delle operazioni sottoposte a ispezione verranno scritti su disco utilizzando il formato di dati NumPy durante l'esecuzione. Se l'opzione pass useDebugInfo è abilitata, le informazioni di debug della località verranno utilizzate, quando disponibili, per identificare in modo univoco i valori dei tensori strumentati (ovvero il pass estrarrà probe_id da NamedLoc(probe_id@<...>) e utilizzerà il formato probe_id.#). In caso contrario, i valori instrumentati verranno indicati nella sequenza crescente: probe1, probe2 e ... consulta interpreter.probe per ulteriori informazioni su come vengono serializzati i dati.

Opzioni

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