-interpreter-instrument-with-probe

Wstawia operacje interpreter.probe w programie StableHLO.

Analizuje program StableHLO i wstawia operację instrumentacji sondy po każdej odpowiedniej operacji (poniżej znajdziesz informacje o tym, jak definiować odpowiednią operację). Instrumentacja służy do wyodrębniania wartości pośrednich z interpretatora referencyjnego StableHLO w celu późniejszego porównania z innymi środowiskami wykonawczymi.

Wszystkie operacje są uważane za odpowiednie do instrumentacji, z wyjątkiem operacji stałych oraz operacji, które nie zwracają wartości tensora (czyli operacje, które zwracają parę lub token, lub nie zwracają żadnej wartości, nie będą instrumentowane). Odpowiednie operacje są instrumentowane niezależnie od poziomu zagnieżdżenia. Oznacza to, że operacje w regionach pętli lub gałęzi również będą monitorowane.

Wartości zwracane z zaimplementowanych operacji są zapisywane na dysku w formacie danych NumPy w miarę ich wykonywania. Jeśli opcja useDebugInfo jest włączona, informacje debugowania lokalizacji będą używane (jeśli są dostępne) do jednoznacznego identyfikowania wartości tensora z instrumentacją (czyli przekształcenie wyekstrahuje probe_id z NamedLoc(probe_id@<...>) i użyje formatu probe_id.#). W przeciwnym razie wartości z instrumentacją będą się odnosić do rosnącej sekwencji: probe1, probe2, … Więcej informacji o serializacji danych znajdziesz w dokumentacji interpreter.probe.

Opcje

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