Code

Make torch xla available on GPU#2176

Code

Merged

muellerzr merged 18 commits intohuggingface:mainhuggingface/accelerate:mainfrom anw90:xla_gpuanw90/accelerate:xla_gpuCopy head branch name to clipboardFeb 14, 2024

+356-150

Commits

Commits on Jan 19, 2024

• Make torch xla available on GPU

  anw90

  committed
  8534c39

  Copy full SHA for 8534c39

• format code

  anw90

  committed
  266a5e0

  Copy full SHA for 266a5e0

• fix documentation build error

  anw90

  committed
  97202e7

  Copy full SHA for 97202e7

• update according to the comments

  anw90

  committed
  eccc542

  Copy full SHA for eccc542

• Replace DistributedType.TPU with DistributedType.XLA

  anw90

  committed
  8ced3ef

  Copy full SHA for 8ced3ef

• make all ut pass

  anw90

  committed
  6b6fed5

  Copy full SHA for 6b6fed5

• format code

  anw90

  committed
  8994b4a

  Copy full SHA for 8994b4a

• update comments

  anw90

  committed
  fe2168b

  Copy full SHA for fe2168b

• skip test

  anw90

  committed
  fb2c3a7

  Copy full SHA for fb2c3a7

• format code

  anw90

  committed
  86d3bce

  Copy full SHA for 86d3bce

• skip FSDPPluginIntegration for torchxla

  anw90

  committed
  e20e1a3

  Copy full SHA for e20e1a3

• bring back custom_sampler_check

  anw90

  committed
  d39fe9a

  Copy full SHA for d39fe9a

• fix ut

  anw90

  committed
  ce7cc39

  Copy full SHA for ce7cc39

Commits on Feb 12, 2024

• Merge branch 'main' into xla_gpu

  muellerzr

  authored
  2924f61

  Copy full SHA for 2924f61

Commits on Feb 14, 2024

• format code

  anw90

  committed
  5683b0a

  Copy full SHA for 5683b0a

• Merge branch 'main' into xla_gpu

  anw90

  committed
  b37697a

  Copy full SHA for b37697a

• format code

  anw90

  committed
  0811bb4

  Copy full SHA for 0811bb4

• Merge branch 'main' into xla_gpu

  anw90

  committed
  a9aadb6

  Copy full SHA for a9aadb6