pytorch/xla

Pull Requests Commits

Implement Fully Sharded Data Parallel (FSDP) in PyTorch XLA (#3431)

ronghanghu committed 3 years ago

Verified 3c83269f

allow disabling layout pinning in optimizer_step and reduce_gradients (#3556)

ronghanghu committed 3 years ago

Verified 6e3992ac

Lower LogSigmoid, LogSigmoidBackward, Elu, and EluBackward (#3539)

wonjoo-wj committed 3 years ago

Verified cc312e27

Delete .torch_pin

JackCaoG committed 3 years ago

Verified 8baddaba

Remove unpool backward (#3537)

JackCaoG committed 3 years ago

Verified 87fda0ac

Update matmul cpp test to check a different counter (#3154)

JackCaoG committed 3 years ago

Verified aa34578e

Lower celu, celu_, selu, and selu_ (#3547)

wonjoo-wj committed 3 years ago

Verified d1de2477

Add ir_builder include since MakeNode moved to this file (#3548)

JackCaoG committed 3 years ago

Verified a3031d64

Start implementation of PjRtComputationClient. (#3463)

will-cromar committed 3 years ago

Verified 9525a8ec

Fix view with no-op view_info (#3540)

JackCaoG committed 3 years ago

Verified c8661df4

change new line to "\n" in XLA_EMIT_STEPLOG (#3546)

ronghanghu committed 3 years ago

Verified 6cc9f3f7

Remove namespaces ir and ops (#3533)

wonjoo-wj committed 3 years ago

Verified 9df591d7

Update libtpu nightly version to 0.1.dev20220413 (#3541)

yeounoh committed 3 years ago

Verified 4131f1ec

Tf update 04 2022 (#3535)

yeounoh committed 3 years ago

Verified bb1e5489

Rotate the nvidia repo key in the base docker image (#3538)

yeounoh committed 3 years ago

Verified f684fd43

Move GetXlaTensorOrCreateForWrappedNumber (#3534)

JackCaoG committed 3 years ago

Verified facc7126

Revert "Start XRT server in a separate process in CircleCI CPU test (#3519)" (#3536)

JackCaoG committed 3 years ago

Verified 093d443b

Move device lock before the execution instead of tensor gathering (#3457)

JackCaoG committed 3 years ago

Verified 5d1c4210

Start XRT server in a separate process in CircleCI CPU test (#3519)

JackCaoG committed 3 years ago

Verified 935b6024

Remove TPU_HOST_BOUNDS setting for TPU single host (#3525)

JackCaoG committed 3 years ago

Verified 9f9db76a

Lower Gelu and GeluBackward (#3531)

wonjoo-wj committed 3 years ago

Verified 89c5dad4

Rename torch_xla::ir Node to XlaNode and Value to XlaValue (#3526)

wonjoo-wj committed 3 years ago

Verified d9dc45fc

Add lazy::shape and lazy::shapefn to XlaNode constructor with some clean up (#3528)

JackCaoG committed 3 years ago

Verified cf7e123c

lowering roll op (#3505)

cicirori committed 3 years ago

Verified 4f6b2833

Add Checkpoint api to use optimization barrier (#3524)

JackCaoG committed 3 years ago

Verified 16498c34

Remove torch_xla::Device (#3495)

wonjoo-wj committed 3 years ago

Verified dbc67604

Track tools.codegen rename to torchgen (#3509)

ezyang committed 3 years ago

Verified 7f9310c6

LTC phase 2 clean up (#3521)

JackCaoG committed 3 years ago

Verified 1f0fae92

Revert "Update tf submodule to 04/07/2022 (#3507)" (#3522)

JackCaoG committed 3 years ago

Verified da2d9887

Revert "Revert "Node hash new (#3514)" (#3517)" (#3518)

JackCaoG committed 3 years ago

Verified f4b64885

Older