pytorch/xla

Pull Requests Commits

JackCaoG committed 1 year ago

6339f37e

set env var for profile

JackCaoG committed 1 year ago

d3006e40

add readme and profile example

JackCaoG committed 1 year ago

ebfca8e3

JackCaoG committed 1 year ago

748df105

Add simple example for how to use torch_xla

JackCaoG committed 1 year ago

433cdc14

add a metrics to track persistent cache loading time (#7039)

JackCaoG committed 1 year ago

Verified c1b745e5

[Pallas] Improve segment_ids API UX (#7037)

alanwaketan committed 1 year ago

Verified 5bbe5c82

[Pallas] Improve FlashAttention segment_ids test case (#7034)

alanwaketan committed 1 year ago

Verified 887d3446

Remove date tag for dev images (#7036)

will-cromar committed 1 year ago

Verified b543dc0e

[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)

wonjoolee95 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)

wonjoolee95 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

Older