PR #29334 Make torch xla available on GPU

Make torch xla available on GPU #29334

amyeroberts merged 7 commits into huggingface:main from yitongh:xla_gpu

ArthurZucker requested a review from

muellerzr 1 year ago

muellerzr approved these changes on 2024-03-07

add USE_TORCH_XLA env

fccdd0d3

rename torch_tpu to torch_xla

55042411

better is_torch_xla_available; fix some fsdp and performance issues

82d0d769

fix format

ce737740

fix bug when pjrt_device is cpu

0d218e23

fix bug

f761c885

yitongh force pushed to f761c885 1 year ago

muellerzr approved these changes on 2024-03-08

muellerzr requested a review from

amyeroberts 1 year ago

amyeroberts commented on 2024-03-08

fix the deprecation handling

1d6223e1

amyeroberts approved these changes on 2024-03-11

amyeroberts merged 873d9bb3 into main 1 year ago

Reviewers

amyeroberts

muellerzr

Assignees

No one assigned

Labels

None yet

Milestone

No milestone