pytorch/xla

Pull Requests Commits

Correct libtpu_storage_path for stable libtpu builds (#8576)

tengyifei committed 344 days ago

36f2a32d

Cherrypick #8524 into r2.6 (#8572)

tengyifei committed 344 days ago

Verified c7902e33

Cherrypick #8558 into r2.6 (#8567)

tengyifei committed 345 days ago

Verified c2e6fd37

Cherrypick #8521 into r2.6 (#8563)

tengyifei committed 345 days ago

Verified 31b2b3b7

Introduce HLO graph bindings (#8564)

rpsilva-aws committed 345 days ago

Verified c23cd63d

Metadata agnostic user computation hash (#8557)

rpsilva-aws committed 348 days ago

Verified 8f5d7bd7

[Computation Hash] Introduce deterministic hash for user computations (#8554)

rpsilva-aws committed 348 days ago

Verified 5ce86092

Add xla autocast support, update autocast APIs in grad checkpointing (#8527)

savitha-aws committed 353 days ago

Verified 28b9b0f8

Cherry pick autocast softmax PR (#8511)

avizon-aws committed 357 days ago

Verified 7d95b391

When modifying IR node, make sure to not lose the read_only bit (#8505) (#8508)

mcuiaws committed 1 year ago

Verified 38ed80e8

xm.save() should not set sync_xla_data=True when sync'ing. (#8484) (#8504)

mcuiaws committed 1 year ago

Verified ef857715

Add xm.xla_device_kind() to return XLA device kind string. (#8493) (#8506)

mcuiaws committed 1 year ago

Verified de9a01ec

Compute and hash buffer_donor_indices for step marker (#8467) (#8503)

mcuiaws committed 1 year ago

Verified cb880ca8

Fix a DDP graph capture issue (#8500)

Chengji Yao committed 1 year ago

Verified 32e3c609

More branch cut changes (#8488)

tengyifei committed 1 year ago

Verified cfe54bf3

Update CI to use PyTorch 2.6 branch (#8487)

tengyifei committed 1 year ago

Verified 142e7027

Update libtpu/xla/jax pin (#8480)

lsy323 committed 1 year ago

Verified 7261c794

Make it easy to switch to nightly libtpu and jax (#8468)

tengyifei committed 1 year ago

Verified 7dd2697a

[AdHoc] Test fix: clear metrics and simplify the replicated counter (#8478)

rpsilva-aws committed 1 year ago

Verified 7ef4a120

Update torch_xla wheels to have the suffix +cxx11 instead of .cxx11 (#8472)

bhavya01 committed 1 year ago

Verified aa56abcc

Register offloading ops as torch ops (#8477)

tengyifei committed 1 year ago

Verified e8720ba6

flash_attention: support also cross attention. (#8427)

dudulightricks committed 1 year ago

Verified 5790092f

[1/N] Use std::string_view (#7297)

cyyever committed 1 year ago

Verified 58be354d

Add autocast support for XlaPatchedLinear (#8421)

aws-nm9 committed 1 year ago

Verified 2021c0fe

[LoweringContext] SPMD propagation (#8471)

rpsilva-aws committed 1 year ago

Verified b068cab9

Modify the deprecation plan for XLA_USE_BF16 (#8474)

rpsilva-aws committed 1 year ago

Verified 18128175

Fix #8475: add libtpu-wheels registry when building docker (#8476)

tengyifei committed 1 year ago

Verified 47f737a7

Add autocast support for einsum (#8420)

aws-nm9 committed 1 year ago

Verified 7b5aca68

Remove Jack and Will from code owner (#8473)

JackCaoG committed 1 year ago

Verified 5c062ea3

Support S32/U32 indices for BWD embedding & Neuron implicit downcast (#8462)

rpsilva-aws committed 1 year ago

Verified 00c0e966

Older