-interpreter-instrument-with-probe

Insère des opérations interpreter.probe dans un programme StableHLO.

Présente un programme StableHLO et insère une opération d'instrumentation de vérification après chaque opération appropriée (voir ci-dessous la définition d'une opération appropriée). L'instrumentation permet d'extraire les valeurs de tenseur intermédiaires de l'interprète de référence StableHLO pour les comparer ultérieurement à d'autres environnements d'exécution.

Toutes les opérations sont considérées comme adaptées à l'instrumentation, à l'exception des opérations constantes, qui ne renvoient aucune valeur de tenseur (c'est-à-dire qu'une opération qui produit un tuple ou un jeton ou aucune valeur de retour ne sera pas instrumentée). Les opérations appropriées seront instrumentées, quel que soit leur niveau d'imbrication. Autrement dit, les opérations dans les régions de boucle/branche seront également instrumentées.

Les valeurs de retour des opérations instrumentées sont écrites sur le disque à l'aide du format de données NumPy au fur et à mesure de leur exécution. Si l'option de passage useDebugInfo est activée, les informations de débogage de localisation seront utilisées, le cas échéant, pour identifier de manière unique les valeurs de tenseur instrumentées (c'est-à-dire que le passage extrait probe_id à partir de NamedLoc(probe_id@<...>) et utilise le format probe_id.#). Sinon, les valeurs instrumentées seront référencées dans la séquence croissante : probe1, probe2, etc. Consultez interpreter.probe pour en savoir plus sur la sérialisation des données.

Options

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