pytorch/xla

Pull Requests Commits

JackCaoG committed 1 year ago

40ca13da

add DDP with SPMD example

JackCaoG committed 1 year ago

dc32271f

Update runner and runner-container-hooks versions (#7058)

mbzomowski committed 1 year ago

Verified f26c35c2

Add simple example for how to use torch_xla (#7048)

JackCaoG committed 1 year ago

Verified ae63cd1e

Add example to support pytorch lightning; misc bug fixes (#7054)

qihqi committed 1 year ago

Verified b64d8a2a

Pin update 20240513 (#7052)

qihqi committed 1 year ago

Verified 2b28ae29

Add option to export FX Node metadata to StableHLO (#7046)

lsy323 committed 1 year ago

Verified 6f392ccc

[Pallas] Add test_pallas_spmd.py to tpu ci (#7045)

alanwaketan committed 1 year ago

Verified 5cb473a2

[FSDPv2] Fix test_fsdp_v2_multi_slice (#7055)

alanwaketan committed 1 year ago

Verified 4a1588c3

[FSDPv2] Support MultiSlice (#7044)

alanwaketan committed 1 year ago

Verified 6f0b61e5

Adding megablox gmm standalone (#6940)

miladm committed 1 year ago

Verified 40f7e1f5

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)

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

Older