pytorch/xla

Pull Requests Commits

Revert "[LTC] Reuse more upstream LazyGraphExecutor data structures (#4303)"

alanwaketan committed 2 years ago

ac7ab5e2

Add note [Note: Re-using upstream TensorImpl] (#4284)

wonjoo-wj committed 2 years ago

Verified 385aa7c3

[LTC] Reuse more upstream LazyGraphExecutor data structures (#4303)

alanwaketan committed 2 years ago

Verified 653df694

Add a forward pass nn model with dynamism test. (#4256)

vanbasten23 committed 2 years ago

Verified 9894c96f

Update torch_xla version for colab from 1.12 to 1.13 (#4295)

vanbasten23 committed 2 years ago

Verified 7a4dcb36

Add a simple gpu doc (#4302)

JackCaoG committed 2 years ago

Verified 632de2eb

[PJRT] Update PJRT release notes with latest changes. (#4269)

will-cromar committed 2 years ago

Verified 9e2fb331

[LTC] Reuse some methods from LazyTensor (#4294)

alanwaketan committed 2 years ago

Verified 16607266

[LTC] Inherits LazyGraphExecutor (#4296)

alanwaketan committed 2 years ago

Verified 036728e9

api to grad base seed as device data (#4293)

shunting314 committed 2 years ago

Verified feba771c

add docs readme back (#4301)

JackCaoG committed 2 years ago

Verified f80a9814

Document environment variables (#4273)

steventk-g committed 2 years ago

Verified 36c53d63

Configure GCB triggers with Terraform (#4267)

will-cromar committed 2 years ago

Verified c9f4c353

Add virtual device metrics (#4279)

steventk-g committed 2 years ago

Verified e4151962

Sharding annotation for empty node (#4292)

yeounoh committed 2 years ago

Verified 5eb9ad46

Update documentation for op related guides (#4283)

wonjoo-wj committed 2 years ago

Verified a1c054da

Lower _native_batch_norm_legit (#4282)

JackCaoG committed 2 years ago

Verified 6fcc8cfc

Fix tf_patches/f16_abi_clang.diff ill-format. (#4278)

vanbasten23 committed 2 years ago

Verified 6a46831e

[PJRT] Create function to spawn threads instead of processes. (#4263)

will-cromar committed 2 years ago

Verified 5cefe63a

Introduce XLAGraphExecutor (#4270)

alanwaketan committed 2 years ago

Verified c20bcdd3

[LTC] Naively inherits LazyTensor (#4271)

alanwaketan committed 2 years ago

Verified f2c87eec

Enable dynamic shape tests on TPUVM (#4259)

JackCaoG committed 2 years ago

Verified 1685a288

Move tensor methods out from XLATensor (#4260)

alanwaketan committed 2 years ago

Verified 4fd09cc8

Clean up some XLATensor unused interfaces (#4257)

alanwaketan committed 2 years ago

Verified 965ca9a2

Fix the documentation link for PJRT. (#4262)

stgpetrovic committed 2 years ago

Verified 4c37d64e

enable nn/mha tests (#4252)

kshitij12345 committed 2 years ago

Verified 6e6bb07e

Fix `non_zero` cast issue (#4243)

JackCaoG committed 2 years ago

Verified 48e12e3d

Update README to remove tpu-node wheels and add gpu 3.8 wheels and dockers (#4251)

JackCaoG committed 2 years ago

Verified d668f2ca

Experimental TPU implementation of DistributedDataParallel (#4193)

will-cromar committed 2 years ago

Verified 803052c1

Fix 1d transpose (#4246)

JackCaoG committed 2 years ago

Verified 0fff6b66

Older