FX tracing improvement #14321
michaelbenayoun
changed the title Fx tracing enhancement FX tracing improvement 4 years ago
changed the way tracing happens, enabling dynamic axes out of the box
07887804
Updated the tests and modeling xlnet
f7306089
Added the non recoding of leaf modules to avoid recording more values…
a146048b
Comments and making tracing work for gpt-j and xlnet
aef4d00a
Refactored things related to num_choices (and batch_size, sequence_le…
f7a69eb6
style fix
b0e3d968
Updated fx to work on PyTorch 1.10
636099b6
Postponed autowrap_function feature usage for later
505333de
style fix
74f74d78
fixed issue
e171ed29
implemented suggestions
6faf2638
Add copyrights
83aedfca
Remove unnecessary file
ee75d025
Merge branch 'master' into fx_tracing_enhancement
bdb1ef85
Fix copies
4f22de3c
Fix issue with add_new_model_like
e566f161
Fix issue with add_new_model_like
a90d319e
Apply suggestions
ae60baf1
Apply suggestions
9ef5813d
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub