-interpreter-instrument-with-probe

SableHLO 프로그램에 interpreter.probe 작업을 삽입합니다.

StableHLO 프로그램을 살펴보고 각 적절한 작업 뒤에 프로브 계측 작업을 삽입합니다(적절한 작업이 정의되는 방식은 아래 참고). 계측은 나중에 다른 런타임과 비교하기 위해 StableHLO 참조 인터프리터에서 중간 텐서 값을 추출하는 데 사용됩니다.

모든 연산은 상수 연산, 텐서 반환 값이 없는 연산(즉, 튜플이나 토큰을 생성하는 연산 또는 반환 값이 없는 연산은 계측되지 않음)을 제외하고 계측에 적합한 것으로 간주됩니다. 적절한 작업은 중첩 수준과 관계없이 계측됩니다. 즉, 루프/브랜치 영역 내부의 작업도 계측됩니다.

계측된 작업은 실행될 때 NumPy 데이터 형식을 사용하여 반환 값을 디스크에 씁니다. useDebugInfo 패스 옵션이 사용 설정된 경우 위치 디버그 정보가 있는 경우 계측된 텐서 값을 고유하게 식별하는 데 사용됩니다(즉, 패스는 NamedLoc(probe_id@<...>)에서 probe_id를 추출하고 probe_id.# 형식을 사용함). 그렇지 않으면 계측된 값은 probe1, probe2 등의 증가하는 순서로 참조됩니다. 데이터가 직렬화되는 방식에 관한 자세한 내용은 interpreter.probe를 참고하세요.

옵션

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