Narzędzie do wyświetlania wykresów
Za pomocą przeglądarki grafów możesz wizualizować strukturę grafu programu XLA. Wyświetla wykres operacji na wysokim poziomie (HLO).
Obsługiwane platformy
TPU: obsługiwane
GPU: obsługiwany
Komponenty interfejsu
Wyświetlaczem wykresu można sterować za pomocą tych opcji:
Menu XLA: moduł, w którym możesz wybrać moduł HLO z listy dostępnych modułów do wizualizacji.
Pole Nazwa operacji, w którym wpisujesz nazwę operacji, aby wybrać ją jako węzeł główny i wizualizować jej sąsiadów. Po wpisaniu nazwy op naciśnięcie klawisza Enter spowoduje wizualizację otaczających ją węzłów.
Selektor szerokości wykresu, który kontroluje maksymalną odległość węzłów od węzła głównego uwzględnionego w wizualizacji.
Selektor metadanych, który służy do przełączania wyświetlania metadanych operacji uwzględnionych przez kompilator (jeśli występują). Pamiętaj, że po zaznaczeniu tego pola musisz ponownie kliknąć „Szukaj w diagramie”.
Pole wyboru Złącz z połączeniem, za pomocą którego możesz zwinąć lub rozwinąć elementy połączonych operacji. Pamiętaj, że po zaznaczeniu tego pola musisz ponownie kliknąć „Wyszukiwanie w diagramie”.
przycisków lub klawiszy powiększania/pomniejszania, choć możesz też użyć myszy;
linki do pobrania wykresu jako plików SVG lub dot.
Za pomocą funkcji wyszukiwania możesz szybko odnajdywać i skupiać się na konkretnych operacjach w potencjalnie dużym grafie.
Aby zablokować panel danych w czasie wykonywania w przypadku konkretnej opcji, kliknij ją prawym przyciskiem myszy.
Korzystanie z Graph Viewer
Domyślnie w przeglądarce grafów nie są wybrane żadne operacje, więc zobaczysz pusty ekran. Wyszukaj interesującą Cię operację, aby ją powiększyć i rozpocząć przechodzenie po diagramie. Często zaczynasz od użycia innego narzędzia (np. OpProfile, aby zidentyfikować operację, która zajmuje najwięcej czasu, lub narzędzia Trace Viewer, aby zidentyfikować przyczynę błędów w przetwarzaniu). Kliknięcie opcji w tych narzędziach spowoduje wyświetlenie bezpośredniego linku do tej samej opcji w przeglądarce Graph.
Gdy najedziesz kursorem na pole (odpowiadające operacji HLO), często zobaczysz wiersz kodu użytkownika odpowiadający operacji, w której zdefiniowano funkcję. Te informacje są dostępne tylko wtedy, gdy te metadane są przekazywane do XProf za pomocą wielu warstw przekształceń, przez które przechodzi Twój program Pythona, aby wygenerować plik wykonywalny akceleratora.
Pamiętaj, że profilujący otrzymuje informacje na poziomie HLO z grafu „optymalizowanego HLO” w miejscu, w którym backend kompilatora może przekształcić go w kod dla konkretnego akceleratora. Przeglądarka grafów i wszystkie narzędzia XProf obsługują tę wewnętrzną reprezentację HLO w ramach XLA, a nie StableHLO ani reprezentację zbliżoną do frameworku ML.
Sam graf jest całkowicie statyczny i jest pobierany bezpośrednio z pliku XLA nawet bez wykonywania programu. Przeglądarka grafów nakłada na niego dodatkowe informacje o czasie wykonywania, aby był bardziej użyteczny. Na przykład po najechaniu kursorem na opcję po lewej stronie interfejsu wyświetli się sekcja, w której widać, ile razy ta opcja została wykonana w czasie trwania profilu i jaki był średni czas jej wykonywania. Oblicza i wyświetla też różne wartości wykorzystania i przepustowości. Do tego celu wykorzystuje statyczną analizę kosztów z XLA w przypadku FLOPS i spożytych bajtów oraz oblicza stawkę na podstawie zmierzonego czasu wykonania.