• Michael Benayoun's avatar
    Fx with meta (#16836) · 2c2a2169
    Michael Benayoun authored
    * Add meta proxy
    
    * Uses meta data to trace data dependent control-flow
    
    * Remove commented class
    
    * Handles torch creating functions
    
    * Added type annotation to fix tracing
    
    * Tracing works for everything but T5 and GPT-J
    
    * Almost all previously supported models pass
    
    * All architectures can be traced except T5
    
    * Intermediate commit to have a trace of the comparison operators for HFProxy
    
    * Everything works, except loss computation
    
    * Everything works
    
    * Removed unused import
    
    * Overriden methods do not use underlying ops (linear and torch.matmul), and model attributes are copied to the traced version
    
    * Fix torch_matmul_override
    
    * Change attributes reference to deepcopy
    
    * Remove breakpoint and add torch_index_override
    
    * Small fix
    
    * Fix typo
    
    * Replace asserts by explicit exceptions
    2c2a2169
fx.py 27.6 KB