pytorch/xla

Pull Requests Commits

Update README.md

ManfeiBai committed 2 years ago

a5ccdfe1

Update README.md

ManfeiBai committed 2 years ago

d5159357

Update Dockerfile

ManfeiBai committed 2 years ago

cab6cf11

Update README.md

ManfeiBai committed 2 years ago

f0f82898

Update README.md

ManfeiBai committed 2 years ago

981bfcb6

Update build_torch_wheels.sh

ManfeiBai committed 2 years ago

28e485b8

Update Dockerfile

ManfeiBai committed 2 years ago

3d7f4eb6

Update build_torch_wheels.sh

ManfeiBai committed 2 years ago

2870d53a

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

Older