pytorch/xla

Pull Requests Commits

alanwaketan committed 2 years ago

59cf84d4

alanwaketan committed 2 years ago

56d62ee3

Remove graph executor parts from XLATensor

alanwaketan committed 2 years ago

f500d1f8

Switch to XLAGraphExecutor and make it work

alanwaketan committed 2 years ago

b4cf3f72

Add some missing interfaces from LazyGraphExecutor

alanwaketan committed 2 years ago

fd93f4a8

Switch to use XLAGraphExecutor

alanwaketan committed 2 years ago

8cc4ee69

Switch to use XLAGraphExecutor

alanwaketan committed 2 years ago

dd9eeecf

Switch to use XLAGraphExecutor

alanwaketan committed 2 years ago

37195ecf

Introduce XLAGraphExecutor singleton

alanwaketan committed 2 years ago

67fa8856

alanwaketan committed 2 years ago

89474eef

Remove wrongly moved methods

alanwaketan committed 2 years ago

b4125401

Make it compiled

alanwaketan committed 2 years ago

90f523d1

CopXLAGraphExecutor out

alanwaketan committed 2 years ago

e81d384f

[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

[LTC] Replace XLA_TIMED with TORCH_LAZY_TIMED (#4250)

alanwaketan committed 2 years ago

Verified fc51c9f2

CI requires cmake>=3.18 to install PyTorch (#4247)

yeounoh committed 2 years ago

Verified 519f5942

Try to make op tests runs on TPUVM (#4244)

JackCaoG committed 2 years ago

Verified bb089b0f

[LTC] Replace XLA_VALUE_METRIC with TORCH_LAZY_VALUE_METRIC (#4245)

alanwaketan committed 2 years ago

Verified 09498a85

Add comment to stream executor build file patch (#4249)

steventk-g committed 2 years ago

Verified e564f825

[LTC] Replace XLA_FN_COUNTER with TORCH_LAZY_FN_COUNTER (#4217)

alanwaketan committed 2 years ago

Verified b969cba3

Companion PR for pytorch/pytorch#89327 (#4242)

Nikita Karetnikov committed 2 years ago

Verified 71c5a005

Older