-interpreter-instrument-with-probe

Fügt interpreter.probe-Vorgänge in ein StableHLO-Programm ein.

Durchläuft ein StableHLO-Programm und fügt nach jedem geeigneten Vorgang einen Probe-Instrumentierungsvorgang ein. Eine geeignete Operation ist unten definiert. Mithilfe der Instrumentierung werden Zwischentensorwerte aus dem StableHLO-Referenzinterpreter extrahiert, um sie später mit anderen Laufzeiten zu vergleichen.

Alle Vorgänge sind für die Instrumentierung geeignet, mit Ausnahme von konstanten Operationen, die keine Tensor-Rückgabewerte haben (d.h. Vorgänge, die ein Tupel, ein Token oder keine Rückgabewerte erzeugen, werden nicht instrumentiert). Geeignete Vorgänge werden unabhängig von ihrer Verschachtelungsebene instrumentiert. Das bedeutet, dass auch Vorgänge in Schleifen-/Abzweigungen erfasst werden.

Die Rückgabewerte von instrumentierten Vorgängen werden bei der Ausführung im NumPy-Datenformat auf die Festplatte geschrieben. Wenn die Option für die Karte/das Ticket useDebugInfo aktiviert ist, werden standortbezogene Informationen zur Fehlerbehebung verwendet, sofern verfügbar, um die instrumentierten Tensorwerte eindeutig zu identifizieren. Das heißt, die Karte/das Ticket extrahiert probe_id aus NamedLoc(probe_id@<...>) und verwendet das Format probe_id.#. Andernfalls werden die instrumentierten Werte in der aufsteigenden Reihenfolge referenziert: probe1, probe2 usw. Weitere Informationen zur Serialisierung von Daten findest du unter interpreter.probe.

Optionen

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