כתבה
arXiv cs.LG ·
GraphMend: תיקון שבירות גרף ב-PyTorch 2
GraphMend: Code Transformations for Fixing Graph Breaks in PyTorch 2
GraphMend הוא טכניקת קומפיילר שמתקנת שבירות גרף בתוכנות PyTorch 2. היא מזהה דפוסים שגורמים לשבירות ומבצעת המרות קוד. GraphMend משפרת ביצועים ומאפשרת גרפים גדולים יותר.
תקציר מקורי באנגליתarXiv:2509.16248v5 Announce Type: replace-cross Abstract: This paper presents GraphMend, a compiler technique that automatically fixes FX graph breaks in PyTorch 2 programs. Although PyTorch 2 introduced TorchDynamo and TorchInductor to enable just-in-time graph compilation, certain code patterns still cause graph breaks that force execution to fall back to Python eager mode, introducing costly CPU-GPU synchronization and reducing optimization opportunities. Our investigation of 195 Hugging Face models reveals that 13.8% of models exhibit graph breaks. GraphMend automatically eliminates fixable breaks through source-level program analysis and transformations. It analyzes AST-level program structure to identify graph-break patterns and applies transformations only when their semantic preser
קרא במקור המקורי
arxiv.org
פתח כתבה מקורית