pytorch/xla

Pull Requests Commits

Update setup.py

ManfeiBai committed 330 days ago

Verified 4cc859b4

Update setup.py

ManfeiBai committed 330 days ago

Verified 0b20c074

Update setup.py

ManfeiBai committed 330 days ago

Verified f2ab3c3d

Update setup.py

ManfeiBai committed 330 days ago

Verified 7c5e37c6

Update setup.py

ManfeiBai committed 330 days ago

Verified 79569e03

Cherrypick #8529 into r2.6 (#8581)

tengyifei committed 331 days ago

Verified 7f75b307

Cherrypick #8576 into r2.6 (#8580)

tengyifei committed 331 days ago

Verified 56e68c6a

Cherrypick #8524 into r2.6 (#8572)

tengyifei committed 332 days ago

Verified c7902e33

Cherrypick #8558 into r2.6 (#8567)

tengyifei committed 332 days ago

Verified c2e6fd37

Cherrypick #8521 into r2.6 (#8563)

tengyifei committed 332 days ago

Verified 31b2b3b7

Introduce HLO graph bindings (#8564)

rpsilva-aws committed 332 days ago

Verified c23cd63d

Metadata agnostic user computation hash (#8557)

rpsilva-aws committed 335 days ago

Verified 8f5d7bd7

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

rpsilva-aws committed 336 days ago

Verified 5ce86092

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

savitha-aws committed 340 days ago

Verified 28b9b0f8

Cherry pick autocast softmax PR (#8511)

avizon-aws committed 344 days ago

Verified 7d95b391

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

mcuiaws committed 357 days ago

Verified 38ed80e8

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

mcuiaws committed 358 days ago

Verified ef857715

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

mcuiaws committed 358 days ago

Verified de9a01ec

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

mcuiaws committed 358 days ago

Verified cb880ca8

Fix a DDP graph capture issue (#8500)

Chengji Yao committed 358 days 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

Older