pytorch/xla

Pull Requests Commits

Add resnet 18 test

JackCaoG committed 2 years ago

67c3367c

Add Training dynamo test for xla

JackCaoG committed 2 years ago

6b48970d

Udpate TF pin to enable `libtpu-nightly==0.1.dev.20221223` (#4429)

yeounoh committed 2 years ago

Verified 10528bf4

Add 1.13 python 3.8 cuda wheel/docker to the doc (#4413)

JackCaoG committed 2 years ago

Verified dfe07ed2

Uncodegen addcmul and addcdiv (#4439)

JackCaoG committed 2 years ago

Verified 93ed11ba

Add more tests for sizeAdd node. (#4258)

vanbasten23 committed 2 years ago

Verified 0062e45a

Add dynamic shape support to sigmoidbackward (#4322)

vanbasten23 committed 2 years ago

Verified f1f9080b

Make dyanmo execute async (#4425)

JackCaoG committed 2 years ago

Verified 53e5d1d8

Fix `xargs` arguments for `patch` in experimental build. (#4437)

will-cromar committed 2 years ago

Verified 570c6d44

Fix TF patch command in experimental build (#4436)

will-cromar committed 2 years ago

Verified 97d52501

Change all size node logical_element_type to be compatible with the underlying XLA device. (#4416)

vanbasten23 committed 2 years ago

Verified 0e696fc7

Fix nightly whl build issue & introduce new knobs in cloudbuild.yaml (#4427)

yeounoh committed 2 years ago

Verified 0986ec50

Rotate GITHUB_TORCH_XLA_BOT_TOKEN and use a new account (#4422)

yeounoh committed 2 years ago

Verified 87ff7d4a

[LTC] XlaBackendImpl::SetRngSeed is not used (#4421)

alanwaketan committed 2 years ago

Verified 3a7c634f

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

Older