pytorch/xla

Pull Requests Commits

Remove GetTensorsFused since we don;t have opbyop anymore

JackCaoG committed 1 year ago

91b76de7

fix typo in TROUBLESHOOTING (#5716)

JackCaoG committed 1 year ago

Verified 4fa727b4

Run decomp before processing (#5713)

qihqi committed 1 year ago

Verified b1123dbb

Add support for `_unsafe_index`. (#5707)

ysiraichi committed 1 year ago

Verified bccbb5aa

Add multi-host GPU support (#5657)

vanbasten23 committed 1 year ago

Verified 6ea99474

Update Troubleshooting with some sanity check example (#5705)

JackCaoG committed 1 year ago

Verified 8f45caef

Filter tensor arguments from traced model. (#5689)

ysiraichi committed 1 year ago

Verified 994f9fb6

pass bundle object to make_tf_function (#5708)

haozha111 committed 1 year ago

Verified 126c9b15

Don't set $TPU_LIBRARY_PATH during import (#5698)

will-cromar committed 1 year ago

Verified 146f2a01

Add --net=host and --shm-size=16g flag to the docker run command in GPU doc. (#5702)

vanbasten23 committed 1 year ago

Verified c9aa49c3

Support async checkpointing through CheckpointManager (#5697)

jonb377 committed 1 year ago

Verified 06ba6e92

Reduce unnecessary tensor allocation in Adam and AdamW (#5700)

baoleai committed 1 year ago

Verified d9523716

Support synchronous saving and loading in CheckpointManager (#5693)

jonb377 committed 1 year ago

Verified 2a45d0d6

Update CI image with dev container image (#5290)

lsy323 committed 1 year ago

Verified e793e864

Open XLA pin update (#5675)

qihqi committed 1 year ago

Verified 418c7518

Add support for unused params (#5694)

qihqi committed 1 year ago

Verified a1bc3c63

Disable xla backend for SPMD (#5690)

jonb377 committed 1 year ago

Verified 5db42bc6

Conditionally set default TPU settings in `__init__.py` (#5696)

will-cromar committed 1 year ago

Verified a98656d2

Fix `masked_fill` broadcasting. (#5688)

ysiraichi committed 1 year ago

Verified 04765eb7

Initial commit for CheckpointManager (#5678)

jonb377 committed 1 year ago

Verified 45245431

Add API to assemble CPU shards to a sharded tensor (#5681)

jonb377 committed 1 year ago

Verified d9a90499

Update artifacts.auto.tfvars for CUDA 12.1 (#5683)

ManfeiBai committed 1 year ago

Verified 2d6c50a4

Add fsdp tests back to GPU CI (#5674)

vanbasten23 committed 1 year ago

Verified b84f7980

Update Dockerfile without tf-nightly (#5684)

ManfeiBai committed 1 year ago

Verified ade17f77

Update pjrt.md (#5679)

ManfeiBai committed 1 year ago

Verified bf470095

Replace pytree_to_str with treespec_dumps (#5526)

apivovarov committed 1 year ago

Verified 4cd48b0b

Revert "Add API to assemble CPU shards to a sharded tensor (#5630)" (#5680)

jonb377 committed 1 year ago

Verified db950b6e

Add API to assemble CPU shards to a sharded tensor (#5630)

jonb377 committed 1 year ago

Verified 3913a77e

Move distributed checkpointing to a subdirectory of experimental (#5656)

jonb377 committed 1 year ago

Verified ac531dcb

Update `setup.py` and README for r2.1 release. (#5669)

will-cromar committed 1 year ago

Verified 43eac0a6

Older