pytorch/xla

Pull Requests Commits

Update dynamo test to be less constrain

JackCaoG committed 1 year ago

09c8b233

Update Openxla-pin to 04/24 (#6975)

ManfeiBai committed 1 year ago

Verified 4b481349

fix pytorch CI after pin update, change test to use assertLessEqual (#6973)

bdhirsh committed 1 year ago

Verified 2a204e9b

Write `torch_xla.version.__torch_gitrev__` to file directly (#6966)

will-cromar committed 1 year ago

Verified 0054ec08

Update GPU readme (#6968)

lsy323 committed 1 year ago

Verified abe090ad

Update readme for 2.3 releae (#6967)

lsy323 committed 1 year ago

Verified 0a204a6b

Enable PagedAttention through Pallas (#6912)

wonjoo-wj committed 1 year ago

Verified 6ed20260

[Doc] Improve docker instructions (#6969)

alanwaketan committed 1 year ago

Verified af74c349

Temporarily ignore torch commit in CI test (#6964)

will-cromar committed 1 year ago

Verified 76f7dd06

Add final 2.3 trigger (#6963)

lsy323 committed 1 year ago

Verified 5369e7d6

[Doc] Update Pallas user guide (#6961)

alanwaketan committed 1 year ago

Verified a7749fa1

update rc14 (#6962)

lsy323 committed 1 year ago

Verified 69eeace3

update torch deps to 2.3 (#6959)

qihqi committed 1 year ago

Verified b5574d83

Create rc13 trigger (#6956)

lsy323 committed 1 year ago

Verified fa090a24

[torch_xla2] Simplify developer setup steps (#6905)

will-cromar committed 1 year ago

Verified 89efd178

Cleanup the code example in the torch_xla2 README. (#6939)

dewitt committed 1 year ago

Verified 6fd448dc

Lower embedding bag forward only (#6951)

bhavya01 committed 1 year ago

Verified 46919a47

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

Older