pytorch/xla

Pull Requests Commits

prepare rc7 for r2.6 with libtpu update

ManfeiBai committed 1 year ago

Verified 90105ab8

Re-introduce "XLA_USE_32BIT_LONG" flag (#8571)

rpsilva-aws committed 1 year ago

Verified 78fa7576

Prepare to make rc6 wheel (#8582)

tengyifei committed 1 year ago

Verified 18efb6dc

[scan] Reduce memory usage (#8562)

tengyifei committed 1 year ago

Verified dc768796

Update terraform providers for tpu-pytorch (#8575)

tengyifei committed 1 year ago

Verified 02602094

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

rpsilva-aws committed 1 year ago

Verified 1d615566

Fix mark_sharding logic (#8578)

yaochengji committed 1 year ago

Verified 14acab7a

Correct libtpu_storage_path for stable libtpu builds (#8576)

tengyifei committed 1 year ago

Verified 79de480f

Prepare to make rc5 wheel (#8573)

tengyifei committed 1 year ago

Verified e513e132

Update iam.auto.tfvars (#8570)

tengyifei committed 1 year ago

Verified 4edbf61d

pin update (#8559)

lsy323 committed 1 year ago

Verified 1c89675e

Set gradient_as_bucket_view=True in test_train_mp_imagenet (#8558)

tengyifei committed 1 year ago

Verified 2fdf721f

fix batch_norm amp autocast (#8556)

yaochengji committed 1 year ago

Verified fda4ecae

[scan] Test we don't recompile under debugging env flags (#8555)

tengyifei committed 1 year ago

Verified 79a7de0c

Introduce HLO graph bindings (#8551)

rpsilva-aws committed 1 year ago

Verified 3784b268

Always run TPU tests on new PRs (#8552)

tengyifei committed 1 year ago

Verified 6963e195

Metadata agnostic user computation hash (#8550)

rpsilva-aws committed 1 year ago

Verified 6e5659ec

Lower cummax op (#8491)

zyy-martin committed 1 year ago

Verified c394d1b5

[torch_xla2] Fix reenabled op info tests (#8548)

dvhg committed 1 year ago

Verified af223d32

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

rpsilva-aws committed 1 year ago

Verified 196cab3f

Revert "fix batch_norm amp autocast" (#8547)

tengyifei committed 1 year ago

Verified 05ca2241

Fix typos in pytorch on xla docs (#8543)

MichalPitr committed 1 year ago

Verified de756865

fix batch_norm amp autocast (#8498)

yaochengji committed 1 year ago

Verified 1cdb1eff

doc change: stablehlo.md (#8544)

qihqi committed 1 year ago

Verified 004845f5

[torch_xla2] Update test_ops.py (#8403)

ManfeiBai committed 1 year ago

Verified df02cacf

fix ab in flash attention (#8540)

ZhiyuLi-goog committed 1 year ago

Verified c673809a

Fully revamp aten embedding bag for JAX backend (#8535)

vyom1611 committed 1 year ago

Verified 5278e7a5

pin update (#8536)

lsy323 committed 1 year ago

Verified d36404c2

Use the same decomp table for eager and export (#8531)

qihqi committed 1 year ago

Verified 0c64c365

[scan] Support different output dtypes (#8530)

tengyifei committed 1 year ago

Verified d3d00ac8

Older