pytorch/xla

Pull Requests Commits

Update eager.md (#7710)

JackCaoG committed 2 years ago

cc9994ab

Add docs for eager mode (#7700)

JackCaoG committed 2 years ago

d8460f8b

Update docs for 2.4 release (#7707)

JackCaoG committed 2 years ago

Verified 55aa8df9

Update README.md for 2.4 release (#7694)

bhavya01 committed 2 years ago

Verified 4d85db19

Add apt-get update in upstream image build (backport #7640) (#7684)

bhavya01 committed 2 years ago

Verified 89ad4feb

Backport Enable eager spmd #7341 and fix eager mode spmd module loading with fsdpv2 #7631 (#7673)

JackCaoG committed 2 years ago

Verified c4daf6ab

cherry-pick Handle random seed for eager mode (#7328) (#7669)

JackCaoG committed 2 years ago

Verified b34abadf

Backport Support eager mode for multi-process training #7327 (#7668)

JackCaoG committed 2 years ago

Verified 6d4c87cc

Cherry-pick Egaerly execute inplace ops if in eager mode #7322 (#7666)

JackCaoG committed 2 years ago

Verified f84f7bec

backport support eager mode with torch.compile #7256 (#7649)

JackCaoG committed 2 years ago

Verified 8ac64257

Backport add torch_xla.experimental.compile for eager mode (#7246) (#7643)

JackCaoG committed 2 years ago

Verified 18b351ff

backport `Enable eager mode for PyTorch/XLA` (#7611)

JackCaoG committed 2 years ago

Verified 7714e782

Update jax to a stable version(0.4.29) for r2.4 (#7627)

bhavya01 committed 2 years ago

Verified 12e0d811

backport #7609: Run paged_attention test with megacore mode only on TPU v4 (#7628)

bhavya01 committed 2 years ago

Verified 85664631

Fetch submodules separately when torch pin is specified (#7616)

bhavya01 committed 2 years ago

Verified 7ed17578

Backport #7329: Avoid OOM when build PyTorch with CUDA enabled (#7618)

bhavya01 committed 2 years ago

Verified 844856d2

Add torch_pin to 2.4 release branch (#7336)

bhavya01 committed 2 years ago

Verified 0a17f095

[backport] Register custom call with CUDA plugin (#7303)

bhavya01 committed 2 years ago

Verified e44787a5

Update cuda_deps.yaml to include CUDA 12.4 (#7244)

bhavya01 committed 2 years ago

Verified 032e332e

[backport][Fori_loop|While_loop] Enable while_loop/fori_loop, add test case (#7157) (#7306)

ManfeiBai committed 2 years ago

Verified 27934620

[backport] Use np.prod instead of np.product (#7301) (#7310)

ManfeiBai committed 2 years ago

Verified a901eb8e

Revert the mul change (#7271) (#7285)

JackCaoG committed 2 years ago

Verified 72961330

Avoid log spam (#7278) (#7284)

JackCaoG committed 2 years ago

Verified fda5828e

Cherry-pick: PJRT plugin defaults (#7249 and #7268) (#7270)

will-cromar committed 2 years ago

Verified f00bd0b9

Backport: Bump jaxversion corresponding to libtpu 0612 (#7262)

vanbasten23 committed 2 years ago

Verified 5622efc4

Backport xla pin update (0612) (#7261)

vanbasten23 committed 2 years ago

Verified da5843c6

Use dest_offsets directly in LoadPlanner (#7243)

jonb377 committed 2 years ago

Verified 62425ebb

Add build trigger for 2.4.0-rc1 release

bhavya01 committed 2 years ago

7ee1bab6

[ZeRO-1] Sync features from r2.1_aws_neuron branch (#7132)

hgt312 committed 2 years ago

Verified 551a76cd

temporarily skip the pjrt test (#7230)

zpcore committed 2 years ago

Verified d0fb59e7

Older