pytorch/xla

Pull Requests Commits

scan documentation (#8631)

tengyifei committed 1 year ago

e47f015a

Cherrypick fix grad accumulation tests on TPUv5/v6 (#8628) (#8634)

tengyifei committed 1 year ago

Verified 8cbe1c84

Update libtpu to 0.0.7.1 (#8630)

tengyifei committed 1 year ago

Verified 5122586f

Update libtpu to 0.0.7.1 (#8630)

tengyifei committed 1 year ago

Verified 25c1d66f

Improve the nprocs value error message to be actionable (#8623)

jeffhataws committed 1 year ago

Verified 6fdbf146

[Cherrypick] use env to skip PJRT initialize (#8609) (#8618)

zpcore committed 1 year ago

Verified a5010889

Introducing experimental gradient accumulation API (#8608)

rpsilva-aws committed 1 year ago

Verified 728f14de

Cherrypick #8562 (#8611)

tengyifei committed 1 year ago

Verified ac568122

Cherrypick #8530 (#8604)

tengyifei committed 1 year ago

Verified a954d927

Cherrypick #8555 (#8605)

tengyifei committed 1 year ago

Verified ff75e1f6

Bump jaxlib to 0.4.38 (#8596)

tengyifei committed 1 year ago

Verified b7096ceb

Improve and refine MLP tests for extensibility and A/B testing (#8590)

rpsilva-aws committed 1 year ago

Verified 19d95da7

Re-introduce "XLA_USE_32BIT_LONG" flag (#8589)

rpsilva-aws committed 1 year ago

Verified fccd3951

[hot fix] Update Libtpu/JAX to newer version for TPU topology (#8585)

ManfeiBai committed 1 year ago

Verified 1dd49698

Cherrypick #8529 into r2.6 (#8581)

tengyifei committed 1 year ago

Verified 7f75b307

Cherrypick #8576 into r2.6 (#8580)

tengyifei committed 1 year ago

Verified 56e68c6a

Cherrypick #8524 into r2.6 (#8572)

tengyifei committed 1 year ago

Verified c7902e33

Cherrypick #8558 into r2.6 (#8567)

tengyifei committed 1 year ago

Verified c2e6fd37

Cherrypick #8521 into r2.6 (#8563)

tengyifei committed 1 year ago

Verified 31b2b3b7

Introduce HLO graph bindings (#8564)

rpsilva-aws committed 1 year ago

Verified c23cd63d

Metadata agnostic user computation hash (#8557)

rpsilva-aws committed 1 year ago

Verified 8f5d7bd7

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

rpsilva-aws committed 1 year ago

Verified 5ce86092

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

savitha-aws committed 1 year ago

Verified 28b9b0f8

Cherry pick autocast softmax PR (#8511)

avizon-aws committed 1 year 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

Older