-interpreter-instrument-with-probe

Вставляет операции интерпретатора.probe в программу StableHLO.

Просматривает программу StableHLO и вставляет операцию зондирования после каждой подходящей операции (см. ниже, как определяется подходящая операция). Инструментарий используется для извлечения промежуточных значений тензора из эталонного интерпретатора StableHLO для последующего сравнения с другими средами выполнения.

Все операции считаются подходящими для инструментирования, за исключением константных операций, операций, которые не имеют каких-либо тензорных возвращаемых значений (т. е. операция, которая создает кортеж или токен или не имеет возвращаемых значений, не будет инструментироваться). Подходящие операции будут инструментированы независимо от уровня их вложенности. То есть операции внутри областей цикла/ветви также будут инструментированы.

Возвращаемые значения инструментированных операций будут записываться на диск в формате данных NumPy по мере их выполнения. Если опция прохода useDebugInfo включена, информация об отладке местоположения будет использоваться, если она доступна, для уникальной идентификации инструментированных значений тензора (т. е. проход извлекает probe_id из NamedLoc(probe_id@<...>) и использует формат probe_id .#). В противном случае инструментальные значения будут упоминаться в возрастающей последовательности: probe1 , probe2 ,... Дополнительную информацию о том, как сериализуются данные, см. interpreter.probe .

Параметры

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