ابزار نمایش گراف
می توانید از Graph Viewer برای تجسم ساختار نمودار برنامه XLA خود استفاده کنید. این نمودار عملیات سطح بالا (HLO) را نشان می دهد.
پلتفرم های پشتیبانی شده
TPU: پشتیبانی می شود
GPU: پشتیبانی می شود
اجزای رابط
Graph Viewer را می توان با گزینه های زیر کنترل کرد:
یک منوی کشویی Select XLA Module، که از آن یک ماژول HLO را از لیست موجود برای تجسم انتخاب می کنید.
یک کادر Op Name، که در آن نام یک عملیات را وارد می‌کنید تا آن را به عنوان گره اصلی انتخاب کنید و همسایگان آن را تجسم کنید. با فشار دادن enter پس از تایپ نام عملیات، گره های اطراف را به تصویر می کشد.
یک انتخابگر Graph Width که حداکثر فاصله گره ها را از گره اصلی که در تجسم گنجانده شده است را کنترل می کند.
یک انتخابگر فراداده، که در صورت وجود، از آن برای تغییر وضعیت نمایش ابرداده عملیاتی که توسط کامپایلر موجود است استفاده می کنید. توجه داشته باشید که پس از انتخاب این کادر، باید دوباره روی «جستجوی نمودار» کلیک کنید.
یک کادر بررسی Merge Fusion که می‌توانید از آن برای جمع کردن یا گسترش اجزای عملیات ذوب شده استفاده کنید. توجه داشته باشید که پس از جابجایی این چک باکس، باید دوباره روی «جستجوی نمودار» کلیک کنید.
دکمه‌ها یا کلیدهای بزرگنمایی/کوچک کردن، اگرچه می‌توانید از ماوس نیز برای بزرگنمایی استفاده کنید.
پیوندهایی برای دانلود نمودار به صورت فایل های SVG یا نقطه ای.
با استفاده از قابلیت جستجو، می توانید به سرعت عملیات خاصی را در نمودار بزرگ بالقوه پیدا کرده و بر روی آنها تمرکز کنید.
برای فریز کردن پانل داده های زمان اجرا در یک عملیات خاص، روی op کلیک راست کنید.
استفاده از Graph Viewer
به طور پیش فرض، هیچ عملیاتی در Graph Viewer انتخاب نمی شود، بنابراین یک صفحه خالی خواهید دید. عملیات مورد علاقه را جستجو کنید تا روی آن بزرگنمایی کنید و شروع به عبور از نمودار کنید. اغلب، شما با یکی از ابزارهای دیگر شروع می‌کنید (مثلاً Op Profile برای شناسایی زمان‌برترین عملیات یا Trace Viewer برای شناسایی علت حباب خط لوله). با کلیک بر روی op در آن ابزارها، یک لینک مستقیم به همان op در Graph Viewer به شما می دهد.
با نگه داشتن ماوس روی یک کادر (نماینده یک عملیات HLO)، اغلب ممکن است خط کد کاربر خود را مطابق با عملیاتی که تابع در آن تعریف شده است ببینید. این اطلاعات تنها در صورتی در دسترس است که این ابرداده از طریق لایه‌های چندگانه تبدیلی که برنامه پایتون شما در مسیر تولید یک شتاب‌دهنده اجرا می‌کند، در اختیار XProf قرار گیرد.
توجه داشته باشید که نمایه‌ساز اطلاعات سطح HLO را از یک نمودار "HLO بهینه شده" دریافت می‌کند، نزدیک به نقطه‌ای که آماده کاهش به کد مخصوص شتاب‌دهنده توسط باطن کامپایلر است. بنابراین، Graph Viewer، و در واقع همه ابزارهای XProf، به جای StableHLO یا نمایشی نزدیک به چارچوب ML، این نمایش HLO داخلی XLA سطح پایین را مدیریت می‌کنند.
در حالی که خود نمودار کاملاً ثابت است و مستقیماً از XLA حتی بدون اجرای برنامه به دست می‌آید، Graph Viewer اطلاعات اضافی زمان اجرا را روی نمودار می‌پوشاند تا آن را مفیدتر کند. به عنوان مثال، با نگه داشتن ماوس روی یک عملیات، بخشی در سمت چپ رابط کاربری ظاهر می‌شود که نشان می‌دهد این عملیات چند بار در طول مدت نمایه اجرا شده است و میانگین زمان اجرا چقدر بوده است. همچنین اعداد مختلف استفاده و پهنای باند را محاسبه و نمایش می دهد. اینها تحلیل هزینه ایستا را از XLA برای FLOPS و بایت های مصرف شده می گیرند و نرخی را بر اساس زمان اجرای اندازه گیری شده محاسبه می کنند.