-interpreter-instrument-with-probe

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

Durchläuft ein StableHLO-Programm und fügt eine Sondeninstrumentierung ein nach jedem geeigneten Vorgang ausführen (siehe unten, -Vorgang definiert ist). Mit Instrumentierung werden Zwischenschritte Tensorwerte aus dem StableHLO-Referenzinterpreter für einen späteren Vergleich mit anderen Laufzeiten.

Alle Vorgänge werden als für die Instrumentierung geeignet angesehen, mit Ausnahme konstanter Vorgänge. Ops, Ops, die keine Tensor-Rückgabewerte haben (d.h. eine Operation, die ein Tupel oder ein Token erzeugt oder es werden keine Rückgabewerte instrumentiert). Es werden unabhängig von der Höhe der Verschachtelung. Das heißt, Vorgänge innerhalb von Loop-/Zweig-Regionen sind ebenfalls instrumentiert ist.

Bei instrumentierten Vorgängen werden ihre Rückgabewerte mit das NumPy-Datenformat verwendet. Wenn useDebugInfo bestanden aktiviert ist, werden Debugging-Informationen für den Standort verwendet, sofern verfügbar identifiziert instrumentierte Tensorwerte eindeutig (d.h. die Karte / das Ticket extrahiert probe_id aus NamedLoc(probe_id@<...>) und verwenden Sie das Format probe_id.#). Andernfalls werden instrumentierte Werte in der zunehmenden Sequenz: probe1, probe2, ... Siehe interpreter.probe für weitere wie Daten serialisiert werden.

Optionen

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