pytorch/xla

Pull Requests Commits

JackCaoG committed 3 years ago

b88d5a09

Skip sync on DeviceData on CollectSynctensors and add metrics test for dynamo+spmd

JackCaoG committed 3 years ago

31b1a08d

[test] Delete `core:lib` in `libxla_computation_client.so` (#5175)

ManfeiBai committed 3 years ago

e42365fe

[OpenXLA] Replace `with_tpu_support` from TSL (#5171)

ManfeiBai committed 3 years ago

b4448baa

Move `third_party/xla_client` to `torch_xla/csrc/runtime` (#5160)

will-cromar committed 3 years ago

0015ea69

Implement SPMDLoadPlanner to enable distributed checkpoint loading (#5130)

jonb377 committed 3 years ago

50821dac

Refine the comment for TensorCollectionBarrier in SyncTensorsGraphInternal (#5166)

JackCaoG committed 3 years ago

36536a7a

[OpenXLA] Replace `tf_cc_shared_object` with `cc_binary` (#5153)

ManfeiBai committed 3 years ago

4f66d6d8

Lower count_nonzero (#5137)

vanbasten23 committed 3 years ago

4da677c7

Throw an error when both operands of torch.add have dynamic dimensions (#5157)

vanbasten23 committed 3 years ago

464b967a

Add utility functions for distributed checkpointing (#5128)

jonb377 committed 3 years ago

c7fe0f98

Fix alias for deprecated `pjrt.rendezvous` (#5155)

will-cromar committed 3 years ago

50431339

Revert "AMP for TPUs v2 (#5148)" (#5154)

JackCaoG committed 3 years ago

a4500e74

AMP for TPUs v2 (#5148)

cowanmeg committed 3 years ago

7b7c3234

Mirrored PR of #5109 (#5150)

lsy323 committed 3 years ago

2f52ca16

Set replicated output false (#5149)

yeounoh committed 3 years ago

0a6e1659

Disable XRT in default CI jobs (#5135)

will-cromar committed 3 years ago

997b2e7c

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

will-cromar committed 3 years ago

0ba21fa2

Update README.md (#5146)

ManfeiBai committed 3 years ago

26951a11

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

JackCaoG committed 3 years ago

e8dc5737

[SPMD] Set replicated output false (#5139)

yeounoh committed 3 years ago

f38248bb

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

ManfeiBai committed 3 years ago

02c41cbb

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

ManfeiBai committed 3 years ago

9195e71b

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

will-cromar committed 3 years ago

ab47dd73

Remove duplicate shard exposure API (#5122)

jonb377 committed 3 years ago

83c48711

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

ManfeiBai committed 3 years ago

d21697da

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

will-cromar committed 3 years ago

7f4d1903

Add load_local_shards to XLAShardedTensor (#5047)

jonb377 committed 3 years ago

864e85df

Update CONTRIBUTING.md (#5119)

JackCaoG committed 3 years ago

46efb75c

Enable dynamism on xla::add (#4868)

vanbasten23 committed 3 years ago

f2123949

Older