-interpreter-instrument-with-probe

Inserta operaciones de interpreter.probe en un programa de StableHLO.

Explica un programa StableHLO y, luego, inserta una operación de instrumentación de sondeo después de cada operación adecuada (consulta a continuación cómo se define una operación adecuada). La instrumentación se usa para extraer los valores de tensor intermedios del intérprete de referencia de StableHLO para una comparación posterior con otros entornos de ejecución.

Todas las operaciones se consideran adecuadas para la instrumentación, excepto las operaciones constantes, que no tienen ningún valor de retorno de tensor (es decir, no se instrumentará una operación que produzca una tupla, un token o ningún valor de retorno). Se instrumentarán las operaciones adecuadas sin importar su nivel de anidación. Es decir, las operaciones dentro de las regiones de bucle o rama también se instrumentarán.

Las operaciones instrumentadas tendrán sus valores de retorno escritos en el disco con el formato de datos NumPy a medida que se ejecutan. Si la opción de pase useDebugInfo está habilitada, se usará la información de depuración de ubicación cuando esté disponible para identificar de forma única los valores de tensor instrumentados (es decir, el pase extraerá probe_id de NamedLoc(probe_id@<...>) y usará el formato probe_id.#). De lo contrario, se hará referencia a los valores instrumentados en la secuencia creciente: probe1, probe2, … Consulta interpreter.probe para obtener información adicional sobre cómo se serializan los datos.

Opciones

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