PR #2176 Make torch xla available on GPU

Make torch xla available on GPU #2176

muellerzr merged 18 commits into huggingface:main from anw90:xla_gpu

muellerzr requested a review from

muellerzr 2 years ago

muellerzr requested changes on 2023-11-27

anw90 force pushed from 5b544459 to b43dc73f 2 years ago

anw90 force pushed from b43dc73f to 45f44d62 2 years ago

anw90 force pushed from 45f44d62 to 72a4322f 2 years ago

muellerzr commented on 2023-12-04

anw90 force pushed from 17553f12 to 7377e418 2 years ago

anw90 force pushed from b0685935 to e1ae723e 2 years ago

anw90 force pushed from 4a7e0230 to dba3950f 2 years ago

anw90 force pushed from dba3950f to 49783032 2 years ago

muellerzr approved these changes on 2024-01-05

muellerzr requested a review from

pacman100 2 years ago

Make torch xla available on GPU

8534c398

format code

266a5e0a

fix documentation build error

97202e75

update according to the comments

eccc5428

Replace DistributedType.TPU with DistributedType.XLA

8ced3efb

make all ut pass

6b6fed5a

format code

8994b4ac

update comments

fe2168b2

skip test

fb2c3a70

format code

86d3bce2

skip FSDPPluginIntegration for torchxla

e20e1a37

bring back custom_sampler_check

d39fe9ae

fix ut

ce7cc39e

anw90 force pushed from 49783032 to ce7cc39e 2 years ago

Merge branch 'main' into xla_gpu

2924f615

format code

5683b0ae

Merge branch 'main' into xla_gpu

b37697ad

format code

0811bb43

Merge branch 'main' into xla_gpu

a9aadb68

muellerzr merged b443be70 into main 2 years ago

Reviewers

muellerzr

pacman100

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

accelerate Make torch xla available on GPU #2176 Merged

Make torch xla available on GPU #2176

accelerate
Make torch xla available on GPU
#2176

Merged