PR #29334 Make torch xla available on GPU

add USE_TORCH_XLA env

yitongh committed 2 years ago

rename torch_tpu to torch_xla

yitongh committed 2 years ago

better is_torch_xla_available; fix some fsdp and performance issues

yitongh committed 2 years ago

fix format

yitongh committed 2 years ago

fix bug when pjrt_device is cpu

yitongh committed 2 years ago

fix bug

yitongh committed 2 years ago

fix the deprecation handling

yitongh committed 2 years ago

transformers Make torch xla available on GPU #29334 Merged