pytorch/xla

Pull Requests Commits

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

ronghanghu committed 4 years ago

Verified f008ec88

use native_layer_norm_backward decomp from pytorch core (#3612)

bdhirsh committed 4 years ago

Verified 08084ea0

Include missing dependencies in setup.py (#3611)

will-cromar committed 4 years ago

Verified 893efc5f

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

JackCaoG committed 4 years ago

Verified 185dd3cd

Make MP_OP_TEST optional (#3605)

JackCaoG committed 4 years ago

Verified 28188b2c

Support type promotion in torch.cat (#3608)

ymwangg committed 4 years ago

Verified 2738899a

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

will-cromar committed 4 years ago

Verified ff2364b6

Enable grad_checkpoint test (#3606)

JackCaoG committed 4 years ago

Verified fde8695f

Full codegen floor and reciprocal (#3600)

wonjoo-wj committed 4 years ago

Verified e2357ba8

Full codegen minimum (#3597)

wonjoo-wj committed 4 years ago

Verified bb43e8a7

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

wonjoo-wj committed 4 years ago

Verified eb7878e3

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

wonjoo-wj committed 4 years ago

Verified 67e4efbc

Full codegen for `logdet` (#3576)

miladm committed 4 years ago

Verified 3c0d68da

Full codegen for `Inverse` (#3575)

miladm committed 4 years ago

Verified e7ebd144

Hotfix cloudbuild bazel update (#3595)

yeounoh committed 4 years ago

Verified 53f4c74b

Wrao XlaData inside of XLABackendData for LTC (#3562)

JackCaoG committed 4 years ago

Verified a67dae3d

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

miladm committed 4 years ago

Verified cc89a31d

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

wonjoo-wj committed 4 years ago

Verified 352dd8c3

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

yeounoh committed 4 years ago

Verified 9a5cfac8

Update libtpu pin to 20220518 nightly (#3586)

yeounoh committed 4 years ago

Verified 8ca18441

Address review comments from #3550 (#3578)

will-cromar committed 4 years ago

Verified 7b3b598f

Remove existing torch_xla::OpList (#3583)

wonjoo-wj committed 4 years ago

Verified 3d24d955

Bazel build issue patch (#3570)

yeounoh committed 4 years ago

Verified bc6471f0

Pin all layout by default and use all_reduce based all_gather when la… (#3568)

JackCaoG committed 4 years ago

Verified 7f96fbf6

Replace XlaValue with upstream lazy::Value (#3572)

wonjoo-wj committed 4 years ago

Verified ec8a7447

Use upstream torch::lazy::Computation (#3580)

JackCaoG committed 4 years ago

Verified 82331cf1

Full codegen for `sgn`, `sign` (#3577)

miladm committed 4 years ago

Verified 041ebf92

codegen for cos and cosh (#3574)

miladm committed 4 years ago

Verified 0bef61c5

Experimental support for single-host TPU in PjRt (#3550)

will-cromar committed 4 years ago

Verified 1ab84187

Revert PR#3450 and use sparse_gather in gather (#3566)

yeounoh committed 4 years ago

Verified 90a525b5

Newer Older