pytorch/xla

Pull Requests Commits

Revert "Index device error (#3632)" (#3640)

JackCaoG committed 3 years ago

Verified 7c99ad06

Index device error (#3632)

bdhirsh committed 3 years ago

Verified 9d5f6f68

delegate sym_sizes to sizes since Pytorch is now using sym_sizes (#3636)

Krovatkin committed 3 years ago

Verified 4f9d3bdc

replace out= lowerings with functional lowerings (#3620)

bdhirsh committed 3 years ago

Verified 9d8706f6

Build nightly torchaudio wheel (#3626)

JackCaoG committed 3 years ago

Verified 943f38f8

Refactor PjRt tests to take advantage of dynamic libtpu loading. (#3628)

will-cromar committed 3 years ago

Verified 3b2bbc8f

More conservative IsSparseGather check to favor dense gather on TPU (#3629)

yeounoh committed 3 years ago

Verified 91095239

Add TPUVM wheel

JackCaoG committed 3 years ago

0c8a82d6

Update nightly version (#3621)

JackCaoG committed 3 years ago

Verified 8ae2ec50

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

Newer Older