pytorch/xla

Pull Requests Commits

Improve test case

alanwaketan committed 1 year ago

faf570e9

Delete customer code

alanwaketan committed 1 year ago

9d6955ee

Enable backward as well

alanwaketan committed 1 year ago

3470ac3b

alanwaketan committed 1 year ago

d81d8699

some minor changes

alanwaketan committed 1 year ago

c28f379d

alanwaketan committed 1 year ago

407e83f2

Handle multiple inplace update input output aliasing (#7023)

JackCaoG committed 1 year ago

Verified e3fc0331

pass TPU_ML_PLATFORM_VERSION env to libtpu (#7021)

zpcore committed 1 year ago

Verified d1235858

Update non_xla attention to properly support paged_attention dynamo code path (#7022)

wonjoo-wj committed 1 year ago

Verified 2bce3f83

Add CI overview to `ci.md` (#7015)

will-cromar committed 1 year ago

Verified aad6a12a

Support export custom op to stablehlo custom call (#7017)

lsy323 committed 1 year ago

Verified 666eccb4

Update XLA/JAX/libtpu pins to 2024/05/02 (#7020)

will-cromar committed 1 year ago

Verified 213b72b9

Update doc push workflow guide and requirement (#7002)

lsy323 committed 1 year ago

Verified 93ce0541

Fix torch.full scalar type (#7010)

JackCaoG committed 1 year ago

Verified 0a54b2b1

Expose python unsafe buffer pointer (#7006)

vanbasten23 committed 1 year ago

Verified 9d84df2a

Support pin pr number in new .torch_pin (#6998)

JackCaoG committed 1 year ago

Verified cbbefa2c

[Pallas] Support segment ids in flash attention (#6943)

alanwaketan committed 1 year ago

Verified 400bd0c9

Build upstream CI image on push to master (#6952)

will-cromar committed 1 year ago

Verified 4883f6fe

Fix q dtype in paged attention kernel (#7011)

wonjoo-wj committed 1 year ago

Verified 865836ad

Fix more opinfo tests (#7008)

qihqi committed 1 year ago

Verified 2907ab30

fix addbmm opinfo (#6993)

qihqi committed 1 year ago

Verified 2399e10f

Tag CI build with git hash (#7003)

will-cromar committed 1 year ago

Verified 77bbf7f3

Clean up workspace before test (#7005)

will-cromar committed 1 year ago

Verified b8f8fa9e

Complain when TensorFlow is installed (#7004)

will-cromar committed 1 year ago

Verified 4a5e238e

Don't fail docs push if there's nothing to commit (#7001)

will-cromar committed 1 year ago

Verified 5f75290e

Move the nightly whl instruction out of the hide area (#7000)

JackCaoG committed 1 year ago

Verified 87329ce0

Re-land: dynamo expand test with view-replay. (#6958)

ysiraichi committed 1 year ago

Verified 42db7096

Manually push to `gh-pages` branch (#6996)

will-cromar committed 1 year ago

Verified d25f4752

Fixes to dynamic_shapes args in test_unbounded_dynamism.py (#6999)

avikchaudhuri committed 1 year ago

Verified 73b915b5

`sudo rm` leftover files in GHA (#6995)

will-cromar committed 1 year ago

Verified 0e032b17

Older