pytorch/xla

Pull Requests Commits

miladm committed 2 years ago

3592f80a

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 3 years ago

Verified 334d1d66

[LTC] Adopt even more LazyTensor interfaces (#4317)

alanwaketan committed 3 years ago

Verified d1748e0f

[LTC] Override LazyGraphExecutor::RunPostOrder() (#4315)

alanwaketan committed 3 years ago

Verified 4ffdfe26

[LTC] Adopt more LazyGraphExecutor virtual interfaces (#4314)

alanwaketan committed 3 years ago

Verified 16e963f8

Correctly configure PJRT execution when nprocs=1. (#4285)

stgpetrovic committed 3 years ago

Verified 35a5dc12

[LTC] Inherits LazyGraphExecutor::DeviceContextArena (#4307)

alanwaketan committed 3 years ago

Verified c570f7f7

Add one more symint expand test. (#4291)

vanbasten23 committed 3 years ago

Verified 6cf15f3d

Add met.celar_all (#4310)

JackCaoG committed 3 years ago

Verified b350f363

[WIP]Disable all failed PyTorch test (#4299)

ManfeiBai committed 3 years ago

Verified e92479a0

Fixing CI (#4313)

alanwaketan committed 3 years ago

Verified 4c7c343c

Older