xla
Enable eager mode for PyTorch/XLA
#7234
Merged

Enable eager mode for PyTorch/XLA #7234

JackCaoG merged 10 commits into master from JackCaoG/eager_mode
JackCaoG
JackCaoG Enable eager mode for PyTorch/XLA
94224d55
JackCaoG JackCaoG added usability
JackCaoG JackCaoG requested a review from alanwaketan alanwaketan 1 year ago
JackCaoG JackCaoG requested a review from will-cromar will-cromar 1 year ago
JackCaoG JackCaoG requested a review from lsy323 lsy323 1 year ago
lsy323
lsy323
lsy323 approved these changes on 2024-06-10
JackCaoG JackCaoG added tpuci
JackCaoG run an example for eager
64b6acf9
JackCaoG add PT_XLA_DEBUG=1 test
4a143784
JackCaoG test eager metrics
b15d8ad8
JackCaoG
qihqi
qihqi approved these changes on 2024-06-11
JackCaoG
JackCaoG move eager mode to the experimental
f9c8765a
JackCaoG Update train_decoder_only_eager.py
d10ca679
JackCaoG add missing file
b04a0f72
JackCaoG linter
a966010f
JackCaoG rename api to eager_mode
1fc06f9d
JackCaoG fix test
e6ef457c
JackCaoG
JackCaoG JackCaoG merged 4d0b94fd into master 1 year ago
JackCaoG JackCaoG deleted the JackCaoG/eager_mode branch 1 year ago
JackCaoG JackCaoG added eager

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone