pytorch/xla

Pull Requests Commits

revert expand test with dynamo (#6950)

bdhirsh committed 1 year ago

Verified 62a2b11c

Make `nms` fallback by default. (#6933)

ysiraichi committed 1 year ago

Verified b06c9c77

Update jinja and sphinx versions to address the vulnearbility (#6946)

yeounoh committed 1 year ago

Verified 9ba844a4

Update XLA pin, 04/19/2024 (#6944)

yeounoh committed 1 year ago

Verified 2ec77062

[Fori_loop|While_loop] Create fori_loop.md (#6942)

ManfeiBai committed 1 year ago

Verified 0417d4d5

Use TPU build for CPU and GPU Python tests (#6921)

will-cromar committed 1 year ago

Verified b2556d6c

[Pallas] Integrate FlashAttention with SPMD (#6935)

alanwaketan committed 1 year ago

Verified 9f2b82dc

Fix profiling in benchmark script (#6934)

vanbasten23 committed 1 year ago

Verified 310f08dc

Add doc explaining how it works (#6937)

qihqi committed 1 year ago

Verified f44e1dfc

[Pallas] Delete the deprecated dynamo code (#6936)

alanwaketan committed 1 year ago

Verified f07c27b2

[SPMD] Introduce high level manual sharding APIs (#6931)

alanwaketan committed 1 year ago

Verified 9b2ac4b1

Add examples for training (#6929)

qihqi committed 1 year ago

Verified 781ee938

Add CONTRIBUTING.md (#6924)

qihqi committed 1 year ago

Verified aee1d37f

[SPMD] Support SPMDShardToFullShape (#6925)

alanwaketan committed 1 year ago

Verified 6e34bbed

Update CUDA 12.1 dev image to python 3.10 (#6928)

will-cromar committed 1 year ago

Verified 42b028e2

[SPMD] Fix test_spmd_full_to_shard_shape for v2 (#6927)

alanwaketan committed 1 year ago

Verified dd726d3f

Add support for few basic mutation ops (#6895)

qihqi committed 1 year ago

Verified dc17f9c0

[SPMD] Support SPMDFullToShardShape (#6922)

alanwaketan committed 1 year ago

Verified 2763248f

Add a devcontainer for gpu dev (#6923)

JackCaoG committed 1 year ago

Verified a7a1357d

Add dynamo expand test. (#6659)

ysiraichi committed 2 years ago

Verified 58a412cb

Build plugin during nightly build (#6917)

will-cromar committed 2 years ago

Verified 5a269d7b

Install libtpu for release dockers (#6919)

alanwaketan committed 2 years ago

Verified ec5615c9

[SPMD] Support manual sharding (#6915)

alanwaketan committed 2 years ago

Verified e5513ff4

fix pixel_shuffle return empty (#6907)

zpcore committed 2 years ago

Verified 43f10715

Build TPU package in CI (#6918)

will-cromar committed 2 years ago

Verified 7375a8ad

Add an option to not generate weights. (#6909)

qihqi committed 2 years ago

Verified fc11bbbf

Add `xla.sync` (#6914)

will-cromar committed 2 years ago

Verified 42452717

[Fori_loop|While_loop] Fori loop wrapped from while_loop (#6850)

ManfeiBai committed 2 years ago

Verified f0354ecc

[torch_xla2] Combined tensor.XLADispatchMode() and functions.XLAFunctionMode() (#6902)

will-cromar committed 2 years ago

Verified cfc70e68

Fix runtime error when run dynamo with a profiler scope (#6913)

JackCaoG committed 2 years ago

Verified 7dfbef02

Newer Older