pytorch/xla

Pull Requests Commits

Mirrored PR of #5109 (#5150)

lsy323 committed 2 years ago

Verified 2f52ca16

Set replicated output false (#5149)

yeounoh committed 2 years ago

Verified 0a6e1659

Disable XRT in default CI jobs (#5135)

will-cromar committed 2 years ago

Verified 997b2e7c

Remove mesh_client.h from PJRT computation client (#5132)

will-cromar committed 2 years ago

Verified 0ba21fa2

Update README.md (#5146)

ManfeiBai committed 2 years ago

Verified 26951a11

Revert "[SPMD] Set replicated output false (#5139)" (#5144)

JackCaoG committed 2 years ago

Verified e8dc5737

[SPMD] Set replicated output false (#5139)

yeounoh committed 2 years ago

Verified f38248bb

[OpenXLA] Delete unused `tf_record_reader.py` from TensorFlow (#5112)

ManfeiBai committed 2 years ago

Verified 02c41cbb

[OpenXLA] Delete unused `@org_tensorflow//tensorflow/core:protos_all_cc` dep from TensorFlow (#5124)

ManfeiBai committed 2 years ago

Verified 9195e71b

Add option to remove XRT from the build (#5114)

will-cromar committed 2 years ago

Verified ab47dd73

Remove duplicate shard exposure API (#5122)

jonb377 committed 2 years ago

Verified 83c48711

Delete unused dep `"@org_tensorflow//tensorflow/core/kernels:data_flow",` from TF (#5093)

ManfeiBai committed 2 years ago

Verified d21697da

Move PJRT Python APIs out of `torch_xla.experimental.*` (#5011)

will-cromar committed 2 years ago

Verified 7f4d1903

Add load_local_shards to XLAShardedTensor (#5047)

jonb377 committed 2 years ago

Verified 864e85df

Update CONTRIBUTING.md (#5119)

JackCaoG committed 2 years ago

Verified 46efb75c

Enable dynamism on xla::add (#4868)

vanbasten23 committed 2 years ago

Verified f2123949

[test] Delete `tensor_shape` and test (#5117)

ManfeiBai committed 2 years ago

Verified 07f21d65

Move TPUCI to TFRT for all tests. (#5098)

JackCaoG committed 2 years ago

Verified efccdad0

[SPMD] Support partial replication. (#4788)

yeounoh committed 2 years ago

Verified 85a4f141

fix a typo (#5113)

wbmc committed 2 years ago

Verified 04ade68e

[SPMD] index local shards correctly in the execution arguments for replication. (#5110)

yeounoh committed 2 years ago

Verified 2078d797

[CI] Pass CXX_ABI to cpp tests (#5107)

malfet committed 2 years ago

Verified 3f076072

Fix reference to ComputationClient::Get() (#5111)

will-cromar committed 2 years ago

Verified 09d6775d

Break dependency cycles within `:computation_client` (#5102)

will-cromar committed 2 years ago

Verified 843c850c

[SPMD] API for getting attributes for global devices (#5091)

khatwanimohit committed 2 years ago

Verified fcf8d438

Checkout Ansible setup at origin's master, not local, untracked master (#5106)

mateuszlewko committed 2 years ago

Verified c9457fd3

Revert "Disable cmake macros (#5021)" (#5105)

malfet committed 2 years ago

Verified 98494504

Add artifacts.auto.tfvars for 2.0 with python 3.10 (#5103)

wonjoo-wj committed 2 years ago

Verified 7a6aa398

Add isvalid check for return futures (#5100)

JackCaoG committed 2 years ago

Verified 3829437b

Parallel tests (#5066)

stgpetrovic committed 2 years ago

Verified 939f8961

Newer Older