pytorch/xla

Pull Requests Commits

JackCaoG committed 1 year ago

6be5202b

handle 2d tensor with 2d mesh case

JackCaoG committed 1 year ago

a86ba432

JackCaoG committed 1 year ago

8c846203

Suport unordered sharding spec for partial replication

JackCaoG committed 1 year ago

8f28425f

Disable test failing on TPU CI (#5314)

JackCaoG committed 1 year ago

Verified cf491f9e

Fix race condition when use > 1 threads to transfer data in parallel loader (#5267)

aws-tianquaw committed 1 year ago

Verified 1dc5af55

Supoort unordered sharding spec correctly (#5305)

JackCaoG committed 1 year ago

Verified 67705065

Add `_sharded_cpu_state_dict` for distributed checkpointing (#5288)

shahyash10 committed 1 year ago

Verified 46a0117b

[SPMD] Allow dumping post optimizations hlo (#5302)

alanwaketan committed 1 year ago

Verified 15ee2191

[SPMD] Support mark_sharding on IRs (#5301)

alanwaketan committed 1 year ago

Verified 6fe5cb98

[SPMD] Avoid recompilations in xs.mark_sharding() (#5300)

alanwaketan committed 1 year ago

Verified caf51687

[SPMD] Add FSDP sharding for test_train_spmd_linear_model.py (#5299)

alanwaketan committed 1 year ago

Verified d01880c5

[SPMD] Use xs.Mesh in test_2d_tensor_3d_mesh (#5295)

khatwanimohit committed 1 year ago

Verified df680da7

Make README more actionable (#5262)

will-cromar committed 1 year ago

Verified c27a493e

Revert pr https://github.com/pytorch/xla/pull/2682 (#5215)

vanbasten23 committed 1 year ago

Verified 8cf85def

[SPMD][Virtual Device]All tensors should be in SPMD:0 C++ device (#5284)

JackCaoG committed 1 year ago

Verified e42fffa0

Fix typo in _test.yml (#5172)

malfet committed 1 year ago

Verified 44033edb

Update inline style code to multiline (#5291)

wonjoolee95 committed 1 year ago

Verified be8718ce

avoid copy proto in PrepareOutputShardingPropagation (#5287)

baoleai committed 1 year ago

Verified 54e2bb69

Lower squeeze.dims (#5286)

wonjoolee95 committed 1 year ago

Verified 865abd74

[Traceable Collecive] Hide token for all_gather (#5232)

alanwaketan committed 1 year ago

Verified 42d92709

Disable torch._dynamo.config.automatic_dynamic_shapes (#5285)

wonjoolee95 committed 1 year ago

Verified ebf9607b

Fix the error where mark_step does not materalize tensors on SPMD:0 (#5281)

JackCaoG committed 1 year ago

Verified cb5944ae

Make it an error if calling sizes() on a dynamic tensor. (#4998)

vanbasten23 committed 1 year ago

Verified d3779cb8

Add `runtime.host_index` (#5283)

will-cromar committed 1 year ago

Verified 68d577ee

Remove or improve several hardcoded TPU test conditions (#5272)

will-cromar committed 1 year ago

Verified b8617f52

Always do build_ext in python setup.py develop (#5273)

qihqi committed 1 year ago

Verified 71e6e9b0

Dsiable test_simple_model_with_different_input_shape since it is curretnly broken by pytorch (#5282)

JackCaoG committed 1 year ago

Verified b799e206

Fix the error when export_torch_model is given a non-tensor (#5277)

qihqi committed 1 year ago

Verified 23e722f6

Delete unused .so file, and .lds files (#5275)

ManfeiBai committed 1 year ago

Verified ba01cf9a

Older