pytorch/xla

Pull Requests Commits

Delete .torch_pin

JackCaoG committed 3 years ago

Verified b821f4b5

Add torch.save(bf16) test (#3579)

JackCaoG committed 3 years ago

Verified 2544253e

Load libtpu when ComputationClient is created. (#3616)

will-cromar committed 3 years ago

Verified 8b6d4fa1

Revert "update tensor indexing to fail on `non_xla_tensor[xla_tensor]` (#3414)" (#3622)

JackCaoG committed 3 years ago

Verified ae61f2f6

Wrap ComputationClient::Computation inside of torch_xla::Computation (#3614)

JackCaoG committed 3 years ago

Verified 41a25f5d

update tensor indexing to fail on `non_xla_tensor[xla_tensor]` (#3414)

bdhirsh committed 3 years ago

Verified 83b5fe8d

XLA SPMD PoC implementation with XRT (#3476)

yeounoh committed 3 years ago

Verified 97ab3b87

Move more test under XLA_SKIP_MP_OP_TESTS flag (#3615)

JackCaoG committed 3 years ago

Verified 5dae54bc

Partially implement LTC BackendInterface (#3571)

JackCaoG committed 3 years ago

Verified 73aa5c5e

Update sparsity check heuristics used in TorchGather (#3610)

yeounoh committed 3 years ago

Verified e6e9f2e6

Allowing `float16` and `bfloat16` computation type and sharding groups in XLA FSDP (#3588)

ronghanghu committed 3 years ago

Verified f008ec88

use native_layer_norm_backward decomp from pytorch core (#3612)

bdhirsh committed 3 years ago

Verified 08084ea0

Include missing dependencies in setup.py (#3611)

will-cromar committed 3 years ago

Verified 893efc5f

Inherit from torch::lazy::LoweringContext (#3585)

JackCaoG committed 3 years ago

Verified 185dd3cd

Make MP_OP_TEST optional (#3605)

JackCaoG committed 3 years ago

Verified 28188b2c

Support type promotion in torch.cat (#3608)

ymwangg committed 3 years ago

Verified 2738899a

Add PjRt tests for MNIST and ImageNet. (#3607)

will-cromar committed 3 years ago

Verified ff2364b6

Enable grad_checkpoint test (#3606)

JackCaoG committed 3 years ago

Verified fde8695f

Full codegen floor and reciprocal (#3600)

wonjoo-wj committed 3 years ago

Verified e2357ba8

Full codegen minimum (#3597)

wonjoo-wj committed 3 years ago

Verified bb43e8a7

Revert "Revert "Full codegen for `logdet` (#3576)" (#3601)" (#3602)

wonjoo-wj committed 3 years ago

Verified eb7878e3

Revert "Full codegen for `logdet` (#3576)" (#3601)

wonjoo-wj committed 3 years ago

Verified 67e4efbc

Full codegen for `logdet` (#3576)

miladm committed 3 years ago

Verified 3c0d68da

Full codegen for `Inverse` (#3575)

miladm committed 3 years ago

Verified e7ebd144

Hotfix cloudbuild bazel update (#3595)

yeounoh committed 3 years ago

Verified 53f4c74b

Wrao XlaData inside of XLABackendData for LTC (#3562)

JackCaoG committed 3 years ago

Verified a67dae3d

updated googletest dependency to its stable version 1.11 pin (#3592)

miladm committed 3 years ago

Verified cc89a31d

Full codegen sin, sinh, and tan (#3581)

wonjoo-wj committed 3 years ago

Verified 352dd8c3

Check if XLA_SANDBOX_BUIDL is unset before applying the bazel patch (#3590)

yeounoh committed 3 years ago

Verified 9a5cfac8

Update libtpu pin to 20220518 nightly (#3586)

yeounoh committed 3 years ago

Verified 8ca18441

Newer Older