"...git@developer.sourcefind.cn:sugon_wxj/megatron-lm.git" did not exist on "840759b8b0d7ca3fb58c811a0e122574dafa62e5"
-
Yitong Huang authored
* 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>
873d9bb3
