-interpreter-instrument-with-probe

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

Présente un programme StableHLO et insère une instrumentation de sonde après chaque opération adaptée (voir ci-dessous pour savoir comment est définie). L'instrumentation est utilisée pour extraire les données valeurs de Tensor provenant de l'interpréteur de référence StableHLO pour une comparaison ultérieure avec 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 ops qui n'ont pas de valeurs renvoyées par un Tensor (c'est-à-dire une opération produit un tuple ou un jeton, ou aucune valeur de retour ne sera instrumentée). Les opérations adaptées seront instrumentées, quel que soit leur niveau ou l'imbrication. Autrement dit, les opérations dans des régions de boucle/branche instrumentée.

Les valeurs de retour des opérations instrumentées sont écrites sur le disque à l'aide de au format de données NumPy lors de leur exécution. Si les useDebugInfo passent est activée, les informations de débogage de localisation seront utilisées identifient de manière unique les valeurs de Tensor instrumentées (la carte extrait probe_id à partir de NamedLoc(probe_id@<...>) et utilisez le format probe_id.#). Sinon, les valeurs instrumentées sont désignées dans les séquence: probe1, probe2, ... voir interpreter.probe pour en savoir plus des informations sur la sérialisation des données.

Options

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