-interpreter-instrument-with-probe

Wstawia operacje interpreter.probe w programie StableHLO.

Przedstawia program StableHLO i wprowadza narzędzia sondy po każdej operacji (poniżej znajdziesz informacje o tym, operacji). Instrumentacja służy do wyodrębniania poziomu pośredniego wartości tensorów z interpretera referencyjnego StableHLO do późniejszego porównania wraz z innymi środowiskami wykonawczymi.

Wszystkie operacje są uznawane za odpowiednie do instrumentacji, oprócz stałych operacje, które nie mają żadnych zwracanych wartości tensorów (operacje op, które generuje krotkę lub token albo żadna zwracana wartość nie jest instrumentowana). Odpowiednie operacje będą obsługiwane niezależnie od ich poziomu zagnieżdżania. Oznacza to, że operacje w regionach pętli/gałęzi również będą instrumentowane.

Wartości zwracane dla operacji instrumentowanych będą zapisywane na dysku za pomocą format danych NumPy podczas wykonywania. Jeśli bilet useDebugInfo jest włączona, dane debugowania lokalizacji będą używane, jeśli będą dostępne dla jednoznacznie identyfikować wartości tensora instrumentowanego (tj. karta zostanie wyodrębniona probe_id z tagu NamedLoc(probe_id@<...>) i użyj formatu probe_id.#). W przeciwnym razie wartości instrumentowane będą przywoływane w rosnącym sekwencja: probe1, probe2, ... Więcej informacji znajdziesz na stronie interpreter.probe na temat sposobu serializacji danych.

Opcje

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