pytorch/xla

Pull Requests Commits

Update setup.py

ManfeiBai committed 2 years ago

Verified 374b2c69

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

Disable ABI on the master branch. (#5641)

vanbasten23 committed 2 years ago

Verified a0a3aba9

lower NativeDropoutBackward (#5642)

JackCaoG committed 2 years ago

Verified d4966dbb

Parallelize CI test runs (#5626)

yeounoh committed 2 years ago

Verified 8412ca7f

Adding link to API_Guide and renaming the doc (#5632)

ssusie committed 2 years ago

Verified b5b902f7

Publish dynamic shape doc. (#5631)

vanbasten23 committed 2 years ago

Verified 5a73f9e3

Older