-interpreter-instrument-with-probe

Insere operações interpreter.probe em um programa StableHLO.

Mostra um programa StableHLO e insere uma instrumentação de sondagem após cada operação adequada (veja abaixo como um é definida). A instrumentação é usada para extrair de tensor do interpretador de referência StableHLO para comparação posterior com outros ambientes de execução.

Todas as operações são consideradas adequadas para instrumentação, exceto as constantes ops que não têm valores de retorno de tensor (ou seja, uma op que produz uma tupla ou um token, ou nenhum valor de retorno não será instrumentado). Operações adequadas serão instrumentadas independentemente do nível de o aninhamento. Ou seja, as operações dentro de regiões de loop/ramificação também serão instrumentado.

Os valores de retorno das operações instrumentadas são gravados em disco com o o formato de dados NumPy conforme eles são executados. Se o useDebugInfo passar estiver ativada, as informações de depuração de local serão usadas quando disponíveis para identificar exclusivamente os valores de tensores instrumentados (ou seja, o cartão extrairá probe_id de NamedLoc(probe_id@<...>) e usam o formato probe_id.#). Caso contrário, os valores instrumentados serão referenciados no sequência: probe1, probe2, ... Consulte interpreter.probe para detalhes informações sobre como os dados são serializados.

Opções

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