pytorch/xla

Pull Requests Commits

alanwaketan committed 2 years ago

26cfb004

Rearrange the test

alanwaketan committed 2 years ago

b3b4bd5f

alanwaketan committed 2 years ago

28a2a6eb

alanwaketan committed 2 years ago

da01f7f0

skip tpu < v4 support

alanwaketan committed 2 years ago

e1629cc5

Cache world_size and ordinal in pjrt

alanwaketan committed 2 years ago

f6ff59f8

alanwaketan committed 2 years ago

a432f999

alanwaketan committed 2 years ago

3aa4ebbf

alanwaketan committed 2 years ago

83f6a16f

Make all_gather single dynamo graph

alanwaketan committed 2 years ago

0272c701

[Distributed] Test if xm.all_reduce produces a single dynamo graph (#4920)

alanwaketan committed 2 years ago

Verified 08574298

[Distributed] Adopts traceable all_reduce (#4915)

alanwaketan committed 2 years ago

Verified 6c7ca993

Update op lowering guide (#4925)

cowanmeg committed 2 years ago

Verified 3c00d573

[Distributed] Hide token from Python for _xla_all_reduce path (#4913)

alanwaketan committed 2 years ago

Verified 4994597f

[Distributed] Move the cached all reduce token to C++ (#4912)

alanwaketan committed 2 years ago

Verified 44c2fa04

[PJRT] Make `rendezvous` optional in `xm.save` (#4907)

will-cromar committed 2 years ago

Verified 3f8abbdc

Apply sharding to all compatible input tensors (#4919)

jonb377 committed 2 years ago

Verified a537020f

Skip test/test_zero1.py for TPU (#4917)

alanwaketan committed 2 years ago

Verified b4b4d0dc

Faster multinomial implementation (#4904)

cowanmeg committed 2 years ago

Verified b1862ccc

Revert "Apply sharding to all compatible input tensors (#4872)" (#4911)

alanwaketan committed 2 years ago

Verified fa059c47

[Distributed] Remove sea-of-devices mode (#4905)

alanwaketan committed 2 years ago

Verified 397ea4d5

Add a test to verify nonzero correctness. (#4903)

vanbasten23 committed 2 years ago

Verified 8af5207d

Pin setuptools version to fix vision build (#4906)

JackCaoG committed 2 years ago

Verified 1e54c235

Update tf pin to 03/2023 (#4840)

wonjoo-wj committed 2 years ago

Verified c1138e88

Lower linalg_vector_norm (#4781)

cowanmeg committed 2 years ago

Verified 11c4d2bd

Enable PJRT C API Client for Intel XPU (#4891)

jzhoulon committed 2 years ago

Verified baea8c55

Update dynamo readme (#4759)

JackCaoG committed 2 years ago

Verified 936b3b84

Apply sharding to all compatible input tensors (#4872)

jonb377 committed 2 years ago

Verified 919f0aee

Update 2.0 to 2.1 in experimental Docker (#4898)

will-cromar committed 2 years ago

Verified 855ba3c3

Update PJRT readme to point to r2.0 during clone (#4894)

JackCaoG committed 2 years ago

Verified 080a70d2

Older