transformers
873d9bb3 - Make torch xla available on GPU (#29334)

Commit
1 year ago
Make torch xla available on GPU (#29334) * add USE_TORCH_XLA env * rename torch_tpu to torch_xla * better is_torch_xla_available; fix some fsdp and performance issues * fix format * fix bug when pjrt_device is cpu * fix bug * fix the deprecation handling --------- Co-authored-by: anw90 <ang868@gmail.com> Co-authored-by: wangang.wa <wangang.wa@alibaba-inc.com>
Author
Parents
Loading