pytorch/xla

Pull Requests Commits

Open XLA pin update (#5675)

qihqi committed 2 years ago

Verified 418c7518

Add support for unused params (#5694)

qihqi committed 2 years ago

Verified a1bc3c63

Disable xla backend for SPMD (#5690)

jonb377 committed 2 years ago

Verified 5db42bc6

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

will-cromar committed 2 years ago

Verified a98656d2

Fix `masked_fill` broadcasting. (#5688)

ysiraichi committed 2 years ago

Verified 04765eb7

Initial commit for CheckpointManager (#5678)

jonb377 committed 2 years ago

Verified 45245431

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

jonb377 committed 2 years ago

Verified d9a90499

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

ManfeiBai committed 2 years ago

Verified 2d6c50a4

Add fsdp tests back to GPU CI (#5674)

vanbasten23 committed 2 years ago

Verified b84f7980

Update Dockerfile without tf-nightly (#5684)

ManfeiBai committed 2 years ago

Verified ade17f77

Update pjrt.md (#5679)

ManfeiBai committed 2 years ago

Verified bf470095

Replace pytree_to_str with treespec_dumps (#5526)

apivovarov committed 2 years ago

Verified 4cd48b0b

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

jonb377 committed 2 years ago

Verified db950b6e

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

jonb377 committed 2 years ago

Verified 3913a77e

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

jonb377 committed 2 years ago

Verified ac531dcb

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

will-cromar committed 2 years ago

Verified 43eac0a6

Clean up some outdated docs (#5663)

JackCaoG committed 2 years ago

Verified bb4ef3a7

Add NEURON device type check to certain operations (#5652)

aws-kingrj committed 2 years ago

Verified 90f7f424

add codegen for masked_fill (#5664)

zpcore committed 2 years ago

Verified c9a13248

Expose replica_id in XLAShard (#5654)

jonb377 committed 2 years ago

Verified fba326e8

Make `ComputationClient::Data` inherit from `torch::lazy::BackendData` (#5653)

will-cromar committed 2 years ago

Verified 745985a9

Print dynamo fallback ops for pt_xla_debug flag (#5662)

JackCaoG committed 2 years ago

Verified f344c9e4

Update dynamo number (#5660)

JackCaoG committed 2 years ago

Verified 347ab7a9

Update artifacts.auto.tfvars for 2.1 (#5659)

ManfeiBai committed 2 years ago

Verified 329879ae

Delete unimplemented methods from `ComputationClient` (#5639)

will-cromar committed 2 years ago

Verified ea5073b4

CI hotfix - Do not parallelize coverage runs (#5648)

yeounoh committed 2 years ago

Verified 1b68776e

Disable AI on master branch. (#5650)

vanbasten23 committed 2 years ago

Verified a6365dc2

Revert "Disable ABI on the master branch. (#5641)" (#5649)

vanbasten23 committed 2 years ago

Verified a1e76b45

lower native_dropout (#5643)

zpcore committed 2 years ago

Verified 2c6e4a77

Remove legacy StreamExecutor TPU client from the build (#5640)

will-cromar committed 2 years ago

Verified 00272c34

Newer Older