-interpreter-instrument-with-probe

הוספת פעולות interpreter.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.