pytorch/xla

Pull Requests Commits

alanwaketan committed 2 years ago

216b3c35

alanwaketan committed 2 years ago

4122a231

Address V3 and remove thread safe guards

alanwaketan committed 2 years ago

ff7885ce

Skip test_zero1 to see if GPU are good

alanwaketan committed 2 years ago

36361ed9

Make token process local and thread safe

alanwaketan committed 2 years ago

a46d6dc5

alanwaketan committed 2 years ago

72a88c99

Remove create token

alanwaketan committed 2 years ago

6a8902f5

Remove token from python

alanwaketan committed 2 years ago

3ca03eac

introduce _set_all_reduce_token

alanwaketan committed 2 years ago

a0c808fb

introduce pybinds

alanwaketan committed 2 years ago

e814d60a

Introduce ResetAllReduceToken

alanwaketan committed 2 years ago

2f753d69

Introduce GetAllReduceToken

alanwaketan committed 2 years ago

51869fc9

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

Fix formatting in `xla_client` (#4892)

will-cromar committed 2 years ago

Verified dc973296

Add html to C++ coverage report (#4886)

steventk-g committed 2 years ago

Verified c98ce753

Lower view_as_complex_copy and view_as_real_copy (#4887)

wonjoo-wj committed 2 years ago

Verified a47a7818

Use 'HEAD' instead of master for PyTroch clone (#4890)

mateuszlewko committed 2 years ago

Verified e5b7cb45

Older