pytorch/xla

Pull Requests Commits

Remove .torch_pin

alanwaketan committed 2 years ago

cfefbf18

Add back ssome XLA counters

alanwaketan committed 2 years ago

ad8696a0

alanwaketan committed 2 years ago

3db6a0cf

alanwaketan committed 2 years ago

f4a550b6

Mark some overrided methods final

alanwaketan committed 2 years ago

3e48ad6b

Inherits torch::lazy::LazyGraphExecutor::DeviceContextArena

alanwaketan committed 2 years ago

12f6c4be

Make DeviceContextArena proctected

alanwaketan committed 2 years ago

8918a3d8

Use torch::lazy::Data inDeviceContext

alanwaketan committed 2 years ago

cfff62bf

Fixing CI (#4313)

alanwaketan committed 2 years ago

Verified 4c7c343c

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

alanwaketan committed 2 years ago

Verified b826def1

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

alanwaketan committed 2 years ago

Verified 07053759

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

Older