-interpreter-instrument-with-probe

Inserta operaciones de intérprete.probe en un programa StableHLO.

Analiza un programa StableHLO y, luego, inserta una instrumentación de sondeo. de las operaciones después de cada operación correspondiente (a continuación, se indica cómo implementar una sola operación de ese tipo). La instrumentación se usa para extraer intermedios Valores del tensor del intérprete de referencia 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 ops, ops que no tienen ningún valor de retorno de tensor (es decir, una op que produce una tupla o un token, o no se instrumentará ningún valor de retorno). Se instrumentarán operaciones adecuadas independientemente de su nivel de durante el período de anidación. Es decir, las operaciones dentro de regiones de bucle o rama también se ejecutarán instrumentados.

Las operaciones instrumentadas tendrán sus valores de retorno escritos en el disco usando el formato de datos NumPy a medida que se ejecutan. Si el useDebugInfo pasa se habilita esta opción, se usará la información de depuración por ubicación cuando esté disponible para identificar de manera única los valores de tensores instrumentados (es decir, el pase extraerá probe_id de NamedLoc(probe_id@<...>) y usa el formato probe_id.#). De lo contrario, los valores instrumentados se mencionarán en la tabla secuencia: probe1, probe2, ... Consulte interpreter.probe para obtener más información sobre cómo se serializan los datos.

Opciones

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