pytorch/xla

Pull Requests Commits

address comments

JackCaoG committed 2 years ago

ede9ed02

Make Dynamo execution async

JackCaoG committed 2 years ago

1bc9d727

created the placeholder

JackCaoG committed 2 years ago

e251bf71

Save the output shape when calling GetGraphHash

JackCaoG committed 2 years ago

52d35f44

Enable xla_sharding_test.py except the virtual device related ones (#4424)

yeounoh committed 2 years ago

Verified 40127959

Disable test_xla_sharding since it breaks CI (#4423)

JackCaoG committed 2 years ago

Verified d67271a6

add missing test (#4420)

JackCaoG committed 2 years ago

Verified f2d1ac1f

Fix worker pool name in GCB config (#4417)

will-cromar committed 2 years ago

Verified 8e4619c0

Initialize logical_element_type before sharding in XLATensor::Data (#4418)

yeounoh committed 2 years ago

Verified 8b8d2898

Update CONTINUE_ON_ERROR if statement in CircleCI bash scripts (#4419)

wonjoo-wj committed 2 years ago

Verified 59f58944

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

vanbasten23 committed 2 years ago

Verified acfb2275

Remove op lowering guide from user facing document (#4412)

JackCaoG committed 2 years ago

Verified e7194d6c

Force fail on all unimplemented methods. (#4409)

vanbasten23 committed 2 years ago

Verified 000324c5

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

Older