pytorch/xla

Pull Requests Commits

add `auto_wrap_policy` into XLA FSDP for automatic wrapping (#4318)

ronghanghu committed 2 years ago

Verified d7d04792

Update README link to section (#4367)

JackCaoG committed 2 years ago

Verified b96c9ffe

Enable dynamo test (#4370)

JackCaoG committed 2 years ago

Verified cad848af

Enable FSDP test on gpu ci (#4372)

JackCaoG committed 2 years ago

Verified f257a203

fix type annotations in FSDP (#4371)

ronghanghu committed 2 years ago

Verified 7819db79

Add a test case for BuildInputOutputAliases (#4329)

alanwaketan committed 2 years ago

Verified 27660d79

Update test_xla_sharding.py (#4364)

wonjoolee95 committed 2 years ago

Verified 0d89710a

Fix warnings in `xrt_init.py` (#4341)

will-cromar committed 2 years ago

Verified 7881f536

Add unsharded computation test for virtual device (#4215)

steventk-g committed 2 years ago

Verified 43116740

Update to clear_all, and add back test (removed in bad merge) (#4362)

steventk-g committed 2 years ago

Verified 18caef5d

switch to pinning layout by default in FSDP to avoid data corruption in PJRT and be consistent with xm.all_reduce (#4359)

ronghanghu committed 2 years ago

Verified b357d87d

Revert "Revert "Use C10_AS_INTARRAYREF_SLOW (#4325)" (#4346)" (#4360)

JackCaoG committed 2 years ago

Verified 547ddf79

[PJRT] Fix TPU local ordinal calculation (#4334)

will-cromar committed 2 years ago

Verified 562a80d9

Temporarily remove MKL related tests (#4345)

steventk-g committed 2 years ago

Verified 09a6e7f1

Store Terraform state in shared GCS bucket. (#4323)

will-cromar committed 2 years ago

Verified fc500b47

Revert "Use C10_AS_INTARRAYREF_SLOW (#4325)" (#4346)

JackCaoG committed 2 years ago

Verified 45542f75

Use sharding hash (#4287)

steventk-g committed 2 years ago

Verified 68e5f1be

Add sharding hash to IR nodes (#4286)

steventk-g committed 2 years ago

Verified 98b788e2

Promote int to float for tan operation (consistent with Pytorch) (#4333)

steventk-g committed 2 years ago

Verified 4f375efc

[SPMD] Use d2d replication via CopyToDevice (#4336)

yeounoh committed 2 years ago

Verified 5eaea4c8

Add test to verify virtual device usage metrics (#4330)

steventk-g committed 2 years ago

Verified 3f015284

fix maxpool2d for XLA dynamo tracing (#4276)

bdhirsh committed 2 years ago

Verified 4013c57b

Update xla_base tag to v0.5 (#4342)

yeounoh committed 2 years ago

Verified 4d14622c

Revert "Revert "Install sympy to ci base docker image (#4337)" (#4339)" (#4340)

JackCaoG committed 2 years ago

Verified 823dd005

Revert "Install sympy to ci base docker image (#4337)" (#4339)

miladm committed 2 years ago

Verified 40bd7d6f

Install sympy to ci base docker image (#4337)

yeounoh committed 2 years ago

Verified a0fa2894

Use C10_AS_INTARRAYREF_SLOW (#4325)

ezyang committed 2 years ago

Verified 1cadc304

Specified vague "Any" type definitions (#4321)

gkroiz committed 2 years ago

Verified 9794392a

Reflect that pjrt now supports `nprocs`. (#4326)

stgpetrovic committed 2 years ago

Verified 2fd2ffee

Adding support for torchrun in xla backend (#3609)

amithrm committed 2 years ago

Verified 334d1d66

Older