pytorch/xla

Pull Requests Commits

add cpu_to_device_transfer_threads

chandrasekhard2 committed 2 years ago

c5790bf8

ResNet optimized

chandrasekhard2 committed 2 years ago

9499d865

Revert "Revert "Adapt to new pytorch grad clipping API (#4480)" (#4484)" (#4486)

wonjoo-wj committed 2 years ago

Verified aff35132

Revert "Adapt to new pytorch grad clipping API (#4480)" (#4484)

wonjoo-wj committed 2 years ago

Verified 3e6bbbbd

Adapt to new pytorch grad clipping API (#4480)

wonjoo-wj committed 2 years ago

Verified eac4e547

Companion PR for https://github.com/pytorch/pytorch/pull/92107 (#4463)

ezyang committed 2 years ago

Verified 5c54655a

Fix shm size (#4470)

steventk-g committed 2 years ago

Verified d636e777

Code-gen LeakyRelu, LeakyReluBackward again (#4468)

lsy323 committed 2 years ago

Verified ba61212b

Enable lowering for upsample_bilinear2d with scale factor (#4464)

lsy323 committed 2 years ago

Verified d74faafc

Enable PJRT C API with `libtpu-nightly` (#4400)

will-cromar committed 2 years ago

Verified c1a9879b

Revert "Codegen LeakyRelu and partially codegen LeakyReluBackward (#4449)" (#4467)

alanwaketan committed 2 years ago

Verified a4d48cb8

Code-gen addcdiv again (#4447)

alanwaketan committed 2 years ago

Verified 9c01c3ca

Allow /dev/shm as sandboxed build base (#4460)

yeounoh committed 2 years ago

Verified f818ba2b

Add `XLA_SANDBOX_BUILD=1` to experimental build (#4455)

will-cromar committed 2 years ago

Verified 28d21204

Rebase test trigger PR (#4459)

steventk-g committed 2 years ago

Verified 255d2fdb

Split virtual device and basic sharding tests (#4451)

steventk-g committed 2 years ago

Verified 5bfe5733

Codegen LeakyRelu and partially codegen LeakyReluBackward (#4449)

lsy323 committed 2 years ago

Verified 5a2bbf36

Add unit test for dynamo+training for pytorch/xla (#4445)

JackCaoG committed 2 years ago

Verified 539ae408

Reenable CPP tests in whl build (#4446)

yeounoh committed 2 years ago

Verified 955c185c

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

Older