pytorch/xla

Pull Requests Commits

Move developer facing doc to Contributing.md

JackCaoG committed 2 years ago

Verified 50e23842

Remove op lowering guide from user facing document

JackCaoG committed 2 years ago

Verified 6028004d

change all size node return value to be compatible with the underlying xla device. (#4410)

vanbasten23 committed 2 years ago

Verified cd73a62d

[SPMD] Use sharding annotation from XLATensor data as ground truth. (#4390)

yeounoh committed 2 years ago

Verified 70109e15

Support release builds in Terraform triggers (#4384)

will-cromar committed 2 years ago

Verified f7678c09

Revert "Update TF pin for `libtpu-nightly==0.1.dev.20221223` compatible commit (#4399)" (#4404)

yeounoh committed 2 years ago

Verified 1858e8bc

Update TF pin for `libtpu-nightly==0.1.dev.20221223` compatible commit (#4399)

yeounoh committed 2 years ago

Verified d43a14a9

Pjrt device attributes (#4403)

darisoy committed 2 years ago

Verified b97f05b3

Use table to represent wheels and dockers in README (#4396)

JackCaoG committed 2 years ago

Verified 73283612

Companion PR for #81762 (#3976)

lezcano committed 2 years ago

Verified 80feefaf

[LTC] Final tweaks on XLATensor and XLAGraphExecutor (#4397)

alanwaketan committed 2 years ago

Verified 1effe54a

Update outdated link to docker instructions (#4395)

JackCaoG committed 2 years ago

Verified b946e9cc

Roll back MKL related changes (#4394)

steventk-g committed 2 years ago

Verified cb1e7d4f

[PT/XLA] Fix timing of ExecuteComputation metrics. (#4327)

stgpetrovic committed 2 years ago

Verified afe7bb6b

JackCaoG committed 2 years ago

Verified 428e299a

Use nomkl to disable mkl optimizations (#4344)

steventk-g committed 2 years ago

Verified 40d3322e

ManfeiBai committed 2 years ago

Verified f919b909

Make Dynamo execution also save the graph when XLA_SAVE_TENSORS_FILE is configured (#4388)

JackCaoG committed 2 years ago

Verified e0603089

Revert "Revert "Add aten::_cdist_forward"" (#4382)

JackCaoG committed 2 years ago

Verified a1d671d6

Fix a typo in test_input_output_aliases.py (#4387)

alanwaketan committed 2 years ago

Verified 6bed07c7

Add GPU doc to readme (#4389)

JackCaoG committed 2 years ago

Verified 14bdea62

Update CI test scripts to allow continuing on test failures (#4385)

wonjoo-wj committed 2 years ago

Verified cfeaedef

Update ci baseimage tag to v0.6 (#4386)

yeounoh committed 2 years ago

Verified b490eda9

Add setup for CI on TPUs (#4018)

steventk-g committed 2 years ago

Verified c4bc972b

Revert "Add aten::_cdist_forward (#4277)" (#4381)

JackCaoG committed 2 years ago

Verified fccbc6a2

Add aten::_cdist_forward (#4277)

ymwangg committed 2 years ago

Verified 3edc88f1

Improve numerical stability of torch.sigmoid (#4311)

ymwangg committed 2 years ago

Verified 453aa651

Add gpu fsdp auto_wrap test (#4378)

JackCaoG committed 2 years ago

Verified 68491026

TF pin for 12/15/2022 (#4324)

ManfeiBai committed 2 years ago

Verified 9cf2565f

Print xla_shape_ in IR too. (#4376)

alanwaketan committed 2 years ago

Verified 86bfeb61

Older