Code

Make torch xla available on GPU#29334

Code

Merged

amyeroberts merged 7 commits intohuggingface:mainhuggingface/transformers:mainfrom yitongh:xla_gpuyitongh/transformers:xla_gpuCopy head branch name to clipboardMar 11, 2024

+120-77

Commits

Commits on Mar 8, 2024

• add USE_TORCH_XLA env

  

  anw90

  authored and

  yitongh

  committed
  fccdd0d

• rename torch_tpu to torch_xla

  

  anw90

  authored and

  yitongh

  committed
  5504241

• better is_torch_xla_available; fix some fsdp and performance issues

  yitongh

  committed
  82d0d76

• fix format

  yitongh

  committed
  ce73774

• fix bug when pjrt_device is cpu

  yitongh

  committed
  0d218e2

• fix bug

  yitongh

  committed
  f761c88

Commits on Mar 11, 2024

• fix the deprecation handling

  yitongh

  committed
  1d6223e