pytorch/xla

Pull Requests Commits

[Pallas] Support FA sm_scale (#7035)

alanwaketan committed 1 year ago

Verified 1c31cde0

Move op dispatching logic into an `Environment` class; and use Mode to capture dispatcher instead of tensor. (#7009)

qihqi committed 1 year ago

Verified 825ba0da

Add a missing case for _unsafe_buffer_pointer tests. (#7026)

vanbasten23 committed 1 year ago

Verified a0063724

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

Rework docs push (#6954)

will-cromar committed 1 year ago

Verified c91171d5

Change name of CI documentation (#6994)

will-cromar committed 1 year ago

Verified 7cc78a68

Update test_export_fx_passes.py (#6972)

avikchaudhuri committed 1 year ago

Verified 75278161

Newer Older