-interpreter-instrument-with-probe

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

Aborda um programa StableHLO e insere uma operação de instrumentação de sonda após cada operação adequada (consulte abaixo como uma operação adequada é definida). A instrumentação é usada para extrair valores de tensor intermediário 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 operações constantes, operações que não têm valores de retorno de tensor (ou seja, uma operação que produz uma tupla ou um token ou nenhum valor de retorno não será instrumentado). As operações adequadas serão instrumentadas, independentemente do nível de anilhamento. Ou seja, as operações dentro de regiões de loop/ramificação também serão instrumentadas.

Os valores de retorno das operações instrumentadas serão gravados no disco usando o formato de dados NumPy conforme são executados. Se a opção de cartão useDebugInfo estiver ativada, as informações de depuração de local serão usadas quando disponíveis para identificar exclusivamente os valores de tensor instrumentados. Ou seja, o cartão vai extrair probe_id de NamedLoc(probe_id@<...>) e usar o formato probe_id.#. Caso contrário, os valores instrumentados serão referenciados na sequência crescente: probe1, probe2, ... Consulte interpreter.probe para mais informações sobre como os dados são serializados.

Opções

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