pytorch/xla

Pull Requests Commits

handle single tensor for method send_to_device_single (#5317)

JackCaoG committed 2 years ago

d507067a

Use `_sharded_cpu_state_dict` functionality to Write Items for SPMD Save Planner (#5315)

shahyash10 committed 2 years ago

8c20cbd6

Enable Some input output aliasing under SPMD (#5320)

JackCaoG committed 2 years ago

9a995403

Disable coverage for now (#5321)

JackCaoG committed 2 years ago

ebb51208

Add tf independent serialization (#5308)

qihqi committed 2 years ago

d9c92bd3

quick refactor on _get_group_assignment (#5318)

JackCaoG committed 2 years ago

37b85187

Fix mismatched GPU docker image in the doc. (#5319)

vanbasten23 committed 2 years ago

def08b43

Support unordered sharding spec for partial replication (#5316)

JackCaoG committed 2 years ago

aac03da0

Supoort unordered sharding spec correctly (#5305)

JackCaoG committed 2 years ago

080fdcf6

Add `_sharded_cpu_state_dict` for distributed checkpointing (#5288)

shahyash10 committed 2 years ago

67ab9750

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

alanwaketan committed 2 years ago

21784cef

[SPMD] Support mark_sharding on IRs (#5301)

alanwaketan committed 2 years ago

ff974276

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

alanwaketan committed 2 years ago

a27382a3

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

alanwaketan committed 2 years ago

af0e0c39

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

khatwanimohit committed 2 years ago

cc4f3043

Make README more actionable (#5262)

will-cromar committed 2 years ago

9a0e24de

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

vanbasten23 committed 2 years ago

07d6f7f5

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

JackCaoG committed 2 years ago

848c00d7

Fix typo in _test.yml (#5172)

malfet committed 2 years ago

e8e66d40

Update inline style code to multiline (#5291)

wonjoo-wj committed 2 years ago

db7c973d

XRT branch cherry-pick 07/10 (#5293)

JackCaoG committed 2 years ago

Verified 4b1742ee

Jack cao g/fix spmd buff is null (#5256)

JackCaoG committed 2 years ago

3a8db635

Fix the linter command in the CI (#5254)

vanbasten23 committed 2 years ago

42db7838

Add nightly build for cuda 12 (#5253)

wonjoo-wj committed 2 years ago

4cedb97a

[Functionalization] Remove view in view_symint (#5231)

alanwaketan committed 2 years ago

Verified 1605972c

Add IAM roles for cloudbuild_editors (#5251)

mateuszlewko committed 2 years ago

Verified 04e892fb

Install `expecttest` in xla_test_job.yaml (#5252)

will-cromar committed 2 years ago

Verified ebd95a65

Remove `test_experimental_pjrt_tpu.py` from TPU CI (#5247)

will-cromar committed 2 years ago

Verified c9f2d91a

remove duplicate autocast_test (#5246)

cowanmeg committed 2 years ago

Verified 458cdce2

AMP for TPUs v3 (#5161)

cowanmeg committed 2 years ago

Verified 347146ec

Older